Java实现 LeetCode 791 自定义字符串排序（桶排序）

791. 自定义字符串排序

字符串S和 T 只包含小写字符。在S中，所有字符只会出现一次。

S 已经根据某种规则进行了排序。我们要根据S中的字符顺序对T进行排序。更具体地说，如果S中x在y之前出现，那么返回的字符串中x也应出现在y之前。

返回任意一种符合条件的字符串T。

示例:
输入:
S = “cba”
T = “abcd”
输出: “cbad”
解释:
S中出现了字符 “a”, “b”, “c”, 所以 “a”, “b”, “c” 的顺序应该是 “c”, “b”, “a”.
由于 “d” 没有在S中出现, 它可以放在T的任意位置. “dcba”, “cdba”, “cbda” 都是合法的输出。
注意:

S的最大长度为26，其中没有重复的字符。
T的最大长度为200。
S和T只包含小写字符。

class Solution {
   public String customSortString(String S, String T) { 
        int[] count = new int[26];
        //先统计T里面的各个字母的数量
        for (char c: T.toCharArray())
            count[c - 'a']++;
 
        StringBuilder ans = new StringBuilder();
        //按照S的顺序把字母放进去
        for (char c: S.toCharArray()) {
            for (int i = 0; i < count[c - 'a']; ++i)
                ans.append(c); 
            count[c - 'a'] = 0;
        }
        //剩下的就是S里面没有的全都加进去
        for (char c = 'a'; c <= 'z'; ++c)
            for (int i = 0; i < count[c - 'a']; ++i)
                ans.append(c);

        return ans.toString();


    }
}