LeetCode 824. 山羊拉丁文

如果单词以元音开头（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。

class Solution {
public:
    string toGoatLatin(string S) {
        unordered_set<char> vowels={'a','A','e','E','i','I','o','O','u','U'};
        istringstream ss(S);
        string temp,ans;
        string str = "ma";
        while(ss >> temp){    
            str += "a";
            if(vowels.find(temp[0])!=vowels.end())     //单词首字母是元音
                ans += temp+str+" ";
            else 
                ans += temp.substr(1)+temp[0]+str+" ";       //单词首字母不是元音
        }
        ans.pop_back();//最后的空格
        return ans;
    }
};