LeetCode:824.山羊拉丁文

题目：

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

源码：

class Solution {
    public String toGoatLatin(String S) {
        char[] chars = {'a', 'e', 'i', 'o', 'u', 
        'A', 'E', 'I', 'O', 'U'};
        Set<Character> set = new HashSet<>();
        for (char c : chars) {
            set.add(c);
        }
        String[] str = S.split(" ");
        for (int i = 0; i < str.length; i++) {
            String s = str[i];
            StringBuilder sb = new StringBuilder();
            // 包含元音开头的单词
            if (set.contains(s.charAt(0))) {
                for (int j = 0; j < s.length(); j++) {
                    sb.append(s.charAt(j));
                }
                sb.append(add(i));
                s = sb.toString();
            } else {
                for (int k = 1; k < s.length(); k++) {
                    sb.append(s.charAt(k));
                }
                sb.append(s.charAt(0));
                sb.append(add(i));
            }
            str[i] = sb.toString();
        }
        StringBuilder sb2 = new StringBuilder();
        for (int m = 0; m < str.length; m++) {
            if (m == str.length - 1) {
                sb2.append(str[m]);
                break;
            }
            sb2.append(str[m]);
            sb2.append(" ");
        }
        return sb2.toString();
    }
    // 此函数表示添加 m 和对应数量 a 的情况
    public String add(int x) {
        StringBuilder sb = new StringBuilder();
        sb.append('m');
        for (int i = 0; i < x + 1 + 1; i++) {
            sb.append('a');
        }
        String str = sb.toString();
        return str;
    }
}