LeetCode 824. 山羊拉丁文（C++、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。

C++

class Solution {
public:
    string toGoatLatin(string S) 
    {
        string res;
        int n=S.length();
        vector<string> tmp;
        int start=0;
        for(int i=0;i<n;i++)
        {
            if(' '==S[i])
            {
                tmp.push_back(S.substr(start,i-start));
                start=i+1;
            }
        }
        tmp.push_back(S.substr(start,n-start));
        for(int i=0;i<tmp.size();i++)
        {
            if('a'==tmp[i][0] || 'e'==tmp[i][0] || 'i'==tmp[i][0] || 'o'==tmp[i][0] || 'u'==tmp[i][0] || 'A'==tmp[i][0] || 'E'==tmp[i][0] || 'I'==tmp[i][0] || 'O'==tmp[i][0] || 'U'==tmp[i][0])
            {
                res+=tmp[i]+"ma";
                res.append(i+1,'a');
            }
            else
            {
                res+=tmp[i].substr(1,tmp[i].size()-1);
                res+=tmp[i][0];
                res+="ma";
                res.append(i+1,'a');
            }
            if(i!=tmp.size()-1)
            {
                res+=' ';
            }
        }
        return res;
    }
};

python

class Solution:
    def toGoatLatin(self, S: str) -> str:
        tmp=S.split()
        res=""
        vec=['a','e','i','o','u','A','E','I','O','U']
        for i in range(len(tmp)):
            if tmp[i][0] in vec:
                res+=tmp[i]+"ma"+'a'*(i+1)
            else:
                res+=tmp[i][1:]
                res+=tmp[i][0]+"ma"+'a'*(i+1)
            if i!=len(tmp)-1:
                res+=' '
        return res

LeetCode 824. 山羊拉丁文（C++、python）

猜你喜欢