competitive_programming/sub_str.py at master · zsmn/competitive_programming · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
def separaString(string):
    string += " "
    listaPalavras = []
    listaSimbolos = []
    palavra = ""
    simbolos = [",",".",";","?",":"," "]
    for letra in string:
        if letra not in simbolos:
            palavra += letra
        else:
            listaSimbolos.append(letra)
            if palavra != "":
                listaPalavras.append(palavra)
                palavra = ""
    return listaPalavras,listaSimbolos

def retiraSubstring(lista,listaSimbolos,substring):
    count = 0
    for palavra in lista:
        if palavra == substring:
            lista.remove(palavra)
            listaSimbolos.remove(listaSimbolos[count])
        else:
            indice_da_letra = 0
            palavraFinal = ""
            parcial = ""
            for letra in palavra:
                if letra == substring[indice_da_letra]:
                    parcial += letra
                    if indice_da_letra + 1 < len(substring):
                        indice_da_letra += 1
                else:
                    if parcial != substring:
                        palavraFinal += parcial
                    indice_da_letra = 0
                    parcial = ""
                    if letra == substring[0]:
                        parcial += letra
                        indice_da_letra += 1
                    else:
                        palavraFinal += letra

        if palavraFinal == "":
            lista.remove(palavra)
            listaSimbolos.remove(listaSimbolos[count])
        else:
            lista[count] = palavraFinal
        count += 1


def juntaString(lista,simbolos):
    texto = ""
    ordem = 0
    for palavra in lista:
        texto += palavra + simbolos[ordem]
        ordem += 1
    return texto


string = input("--> ")
substring = input("--> ")
lista,simbolos = separaString(string)
retiraSubstring(lista,simbolos,substring)
print(juntaString(lista,simbolos))