使用嵌套循环优化python代码

Question

这是一个hackerrank问题，我在这里有点菜鸟。 5/15 测试用例正在工作，但是当要分析的字符串太大时，测试用例由于超时而被终止。任何指导将不胜感激。

问题 - 给两个玩家相同的字符串，. 两个玩家都必须使用字符串的字母来创建子字符串。 斯图尔特必须造以辅音开头的单词。 凯文必须以元音开头的单词。 如果一个子串出现x次，则构成+x分。 当两个玩家都完成所有可能的子字符串时，游戏结束。

def minion_game(string):
    vowels = {'A', 'E', 'I', 'O', 'U'}
    strlen = len(string)
    Stuart, Kevin = 0, 0
    words = set([])
    for i in range(strlen):
        for j in range(i, strlen):
            temp_word = string[i:j + 1]
            temp_word_length = j-i+1
            if temp_word not in words:
                words.add(temp_word)
                no_of_search = strlen - temp_word_length + 1
                score = sum(temp_word == string[x : x + temp_word_length] for x in range(no_of_search))
                if string[i] in vowels:
                    Kevin += score
                else:
                    Stuart += score
    if Stuart > Kevin:
        print('Stuart', Stuart)
    elif Kevin > Stuart:
        print('Kevin', Kevin)
    else:
        print('Draw')

Answer 1

我只是看了问题来源。

这可能是你能做到的最短的方法。

def minion_game(string):
    vowels = ['A', 'E', 'I', 'O', 'U']
    kevin = 0
    stuart = 0
    for i in range(len(string)):
        if s[i] in vowels:
            kevin += len(s)-i
        else:
            stuart += len(s)-i

    if stuart > kevin:
        print('Stuart '+ str(stuart))
    elif kevin > stuart:
        print('Kevin ' + str(kevin))
    else:
        print('Draw')

s = input()
minion_game(s)

这部分遍历输入字符串中的每个字母，并检查它是否是元音。

for i in range(len(string)):
    if s[i] in vowels:

HackerRank 会自动为您提供以下信息，如果您确实检查了它的声明位置。

s = input()
minion_game(s)