leetcode 824. 山羊拉丁文(python)

题目链接

题目描述：

给定一个由空格分割单词的句子 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。

解题思路：
用列表yuan保存元音字母aeiou
用.split（）属性将字符串拆分成列表形式
对列表中的每一个单词，验证其第一个字母的小写形式ss[i][0].lower()是否为元音（是否在保存元音的列表中）
若在，说明首字母为元音，用字符串拼接的方法在该单词后加上’ma’，以及索引i+1个a
若不在，说明首字母为辅音，ss[i][1:]+ss[i][0]用切片方法实现将单词第一个字母放在最后，并用字符串拼接的方法在该单词后加上’ma’，以及索引i+1个a

p.s.由于加ma和索引i+1个a是所有单词都要进行的，所以可以不在判断语句中进行

class Solution:
    def toGoatLatin(self, S: str) -> str:
        ss=S.split()
        yuan=['a','e','i','o','u']
        for i in range(len(ss)):
            if ss[i][0].lower() in yuan:#首字母是元音
                ss[i]=ss[i]+'ma'+'a'*(i+1)
            else:
                ss[i]=ss[i][1:]+ss[i][0]+'ma'+'a'*(i+1)
        return ' '.join(ss)