leetcode - 824 - 山羊拉丁文

class Solution:
    def toGoatLatin(self, S):
        """
        :type S: str
        :rtype: str
        """
        
        vowel = ["a","e","i","o","u","A","E","I","O","U"]
        s = S.split()
        for i in range(len(s)):
            if s[i][0] in vowel:
                s[i] += "ma"
            else:
                s[i] = s[i][1:] + s[i][0]
                s[i] += "ma"
            s[i] += "a"*(i+1)
        return " ".join(s)