Leetcode 791. 自定义字符串排序 791. Custom Sort String

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) {
        map<char,int> m;
        string res;
        for(int i=0;i<T.size();i++){      //对T计数，每个字母出现的次数
            if(m.count(T[i])) m[T[i]]++;
            else m[T[i]]=1;
        }
        for(int i=0;i<S.size();i++){      //按照S的顺序，输出每个字母对应的次数个
            if(m.count(S[i])){
                int k=m[S[i]];
                m[S[i]]=0;                //有S中没有出现的，所以将计算过的置0
                while(k>0){
                    res+=S[i];
                    k--;
                }
            }
        }
        for(int i=0;i<26;i++){            //不想遍历map，就遍历26个字母吧；
            char c='a'+i;                 //现在还不是零的，是S中没有的，直接加在后面
            if(m.count(c)){
                int k=m[c];
                while(k>0){
                    res+=c;
                    k--;
                }
            }
        }
        return res;
    }
};