824. 山羊拉丁文

 

class Solution(object):
    def toGoatLatin(self, S):
        """
        :type S: str
        :rtype: str
        """
        # 定义元音字母集合
        vowel = ['a', 'e', 'i', 'o', 'u', 'A', 'E', 'I', 'O', 'U']
        # 用空格分割S
        words = S.split()
        # 返回值
        res = ""
        # 遍历S中的所有单词
        for i, word in enumerate(words):
            # 每个单词拼接时用空格隔开
            res += " "
            if word[0] in vowel:
                res = res + word + "ma" + 'a' * (i + 1)
            else:
                res = res + word[1:] + word[0] + "ma" + 'a' * (i + 1)
        # 截掉首位的空格
        return res[1:]


if __name__ == '__main__':
    solution = Solution()
    print(solution.toGoatLatin("I speak Goat Latin"))