Python Leetcode(824.山羊拉丁文)

Python Leetcode(824.山羊拉丁文)

给定一个由空格分割单词的句子 S。每个单词只包含大写或小写字母。

我们要将句子转换为 “Goat Latin”（一种类似于 猪拉丁文 - Pig Latin 的虚构语言）。

山羊拉丁文的规则如下：

如果单词以元音开头（a, e, i, o, u），在单词后添加"ma"。
例如，单词"apple"变为"applema"。

如果单词以辅音字母开头（即非元音字母），移除第一个字符并将它放到末尾，之后再添加"ma"。
例如，单词"goat"变为"oatgma"。

根据单词在句子中的索引，在单词最后添加与索引相同数量的字母'a'，索引从1开始。
例如，在第一个单词后添加"a"，在第二个单词后添加"aa"，以此类推。

返回将 S 转换为山羊拉丁文后的句子。

示例 1:

输入: “I speak Goat Latin”
输出: “Imaa peaksmaaa oatGmaaaa atinLmaaaaa”

示例 2:

输入: “The quick brown fox jumped over the lazy dog”
输出: “heTmaa uickqmaaa rownbmaaaa oxfmaaaaa umpedjmaaaaaa overmaaaaaaa hetmaaaaaaaa azylmaaaaaaaaa ogdmaaaaaaaaaa”

说明:

S 中仅包含大小写字母和空格。单词间有且仅有一个空格。
1 <= S.length <= 150。

Solution:(建立元音字母列表，对单词的首字母进行判断，不是元音字母的话利用字符串的切片将首字母移到最后，然后加上(i+1)*a)

class Solution(object):
    def toGoatLatin(self, S):
        """
        :type S: str
        :rtype: str
        """
        vowel_words = ['a', 'e', 'i', 'o', 'u', 'A', 'E', 'I', 'O', 'U']
        s_list = S.split(' ')
        result = []
        
        for i, each in enumerate(s_list):
            if each[0] in vowel_words:
                result_words = each + 'ma'
            else:
                result_words = each[1:] + each[0] + 'ma'
            result_words += (i+1) * 'a'
            result.append(result_words)
        
        return ' '.join(result)

solution = Solution()
print(solution.toGoatLatin("I speak Goat Latin"))

Imaa peaksmaaa oatGmaaaa atinLmaaaaa