【Python学习】-Hackrank挑战-字符串-小黄人游戏

题目：小黄人游戏--The Minion Game

题目要求：

对于输入的字符串，计算所有可能的子字符串中，以辅音字母开头的给Stuart加分，元音字母开头的给Kevin加分。然后比较两者的总分，输出胜者或者平局

解题代码：

def minion_game(string):
    vowels = {'A', 'E', 'I', 'O', 'U'}
    stuart_score = 0
    kevin_score = 0
    n = len(string)
    
    for i in range(n):
        char = string[i]
        if char.upper() in vowels:
            kevin_score += (n - i)
        else:
            stuart_score += (n - i)
    
    if stuart_score > kevin_score:
        print(f'Stuart {stuart_score}')
    elif kevin_score > stuart_score:
        print(f'Kevin {kevin_score}')
    else:
        print('Draw')
​
if __name__ == '__main__':
    s = input().strip()
    minion_game(s)