1394. 山羊拉丁文

描述

给定句子S，其由用空格分隔的单词组成。 每个单词仅包含小写和大写字母。

我们想将句子转换为“Goat Latin”（一种类似于Pig Latin的伪造语言）。

Goat Latin的规则如下：

如果一个单词以元音（a，e，i，o或u）开头，则在单词的末尾附加“ma”。
例如，“apple”这个词就变成了“applema”。

如果一个单词以辅音（即不是元音）开头，则删除第一个字母并追加它到最后，然后添加“ma”。
例如，“山羊”这个词就变成了“oatgma”。

在句子的每个单词的末尾添加一个字母'a'，从1开始。
例如，第一个单词将“a”添加到结尾，第二个单词将“aa”添加到结尾，依此类推。

返回将从S到Goat Latin的转换后的最终语句。

• S仅包含大写，小写和空格。 每个单词之间只有一个空格。
• 1 <= S.length <= 150。

样例

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

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

public class Solution {
    /**
     * @param S: 
     * @return: nothing
     */
    public String  toGoatLatin(String S) {
        String[] words = S.split(" ");
        StringBuilder sb = new StringBuilder();
        
        for (int i = 0; i < words.length; ++i) {
            if (sb.length() > 0) {
                sb.append(" ");
            }
            sb.append(normalize(words[i]) + appendAs(i + 1));
        }
        
        return sb.toString();
    }
    
    private String appendAs(int len) {
        StringBuilder sb = new StringBuilder();
        
        for (int i = 0; i < len; ++i) {
            sb.append('a');
        }
        return sb.toString();
    }
    
    private String normalize(String s) {
        HashSet<Character> vowels = new HashSet();
        vowels.add('a');
        vowels.add('o');
        vowels.add('e');
        vowels.add('u');
        vowels.add('i');
        
        String result = "";
        if (s.length() == 0) {
            return "";
        }
        
        if (vowels.contains(s.toLowerCase().charAt(0))) {
            result = s + "ma";
        } else {
            result = s.substring(1, s.length()) + s.charAt(0) + "ma";
        }
        return result;
    }
}