Python数据结构与算法（poj）：P01035——拼写检查

OpenJudge - 01035:拼写检查

        这道题主要考察对于输入单词的判断和处理，时间卡的相对较紧。

        本来想直接打出字典中每个单词对应的变换后的词，存到集合里面，这样判断会快很多，结果MLE。。。（其实后面用sys改了一下，降低了内存，但是还是会远远TLE，说明这种方法是不可取的）

        于是对于每个输入的单词进行判断，这样的优势是（在很多情况下）进行了剪枝，从而可以大大降低时间复杂度。

上代码~

def judge(word, check):
    n, m = len(word), len(check)
    if n == m:
        for k in range(n):
            if word[:k] + word[k + 1:] == check[:k] + check[k + 1:]:
                return True
    elif n == m + 1:
        for k in range(n):
            if word[:k] + word[k + 1:] == check:
                return True
    elif n+1 == m:
        for k in range(m):
            if word == check[:k] + check[k+1:]:
                return True
    return False

my_dict = []
while True:
    word = input()
    if word == '#':
        break
    my_dict.append(word)

while True:
    letter = input()
    if letter == '#':
        break
    if letter in my_dict:
        print(f'{letter} is correct')
    else:
        print(f'{letter}: ', end='')
        for word in my_dict:
            if judge(word, letter):
                print(word, end=' ')
        print()