824. 山羊拉丁文（Python）

题目

难度：★★☆☆☆
类型：字符串

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

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

山羊拉丁文的规则如下：

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

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

根据单词在句子中的索引，在单词最后添加与索引相同数量的字母'a'，索引从1开始。
例如，在第一个单词后添加"a"，在第二个单词后添加"aa"，以此类推。
返回将 S 转换为山羊拉丁文后的句子。

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

示例

示例 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"

解答

1. 由于句子中只有空格和字母，将输入句子使用str.split()拆分成单词；

2. 对每个单词单独处理，根据首字母是否为元音按照题目要求进行不同的转换；

3. 将每一个转换好的单词用空格连接起来，注意如果句子最后有一个空格应该删去。

class Solution:
    def toGoatLatin(self, S):
        vowels = ['a', 'e', 'i', 'o', 'u', 'A', 'E', 'I', 'O', 'U']
        words = S.split()
        res = ''
        for i, word in enumerate(words):
            if word[0] in vowels:
                res += word+'ma'+'a'*(i+1) + ' '
            else:
                res += word[1:]+word[0]+'ma'+'a'*(i+1) + ' '
        return res[:-1]

python知识补充：

enumerate迭代器可以不仅可以获得列表中的元素，同时会获得元素所在的下标，因为题目中有用下标，所以我们使用enumerate。

将列表中的字符串用空格连接起来，除了使用字符串+空格的形式，更方便的是使用join方法，" ".join(word_list)。

上述代码的紧凑写法：

class Solution:
    def toGoatLatin(self, S):
        vowels = ['a', 'e', 'i', 'o', 'u', 'A', 'E', 'I', 'O', 'U']
        return ' '.join([word+'ma'+'a'*(i+1) + ' ' if word[0] in vowels else word[1:]+word[0]+'ma'+'a'*(i+1) + ' ' for i, word in enumerate(S.split())])

如有疑问或建议，欢迎评论区留言~