791. Custom Sort String

一拿到这道题，我首先想的是用binary search做，但是因为结果String的长度不是固定的，用binary search很难实现。所以我写了第一个brute force的算法，时间复杂度是O(n2), 效率很低，beat 5%，而且很容易考虑不到edge case而出错。

    Map<Character, Integer> map = new HashMap<>();

    public String customSortString(String order, String s) {
        for (int i = 0; i < order.length(); i++) {
            map.put(order.charAt(i), i);
        }
        StringBuilder res = new StringBuilder();
        res.append(s.charAt(0));
        for (int i = 1; i < s.length(); i++) {
            char c = s.charAt(i);
            if (!map.containsKey(c))
                res.append(c);
            else
                search(res, c, map);
        }
        return res.toString();
    }

    private void search(StringBuilder s, char c, Map<Character, Integer> map) {
        for (int i = 0; i < s.length(); i++) {
            char temp = s.charAt(i);
            if (!map.containsKey(temp) || map.get(temp) < map.get(c))
                continue;
            else if (map.get(temp) >= map.get(c)) {
                s.insert(i, c);
                return;
            }
        }
        s.append(c);
    }

这道题是求字符串里的对字符的sort，我们还可以利用bucket sort来做，首选把s放在bucket，再用order做sort，最后再把那些没有在order里面的字符加到字符串上即可，时间复杂度O(n), 算法如下：

    public String customSortString(String order, String s) {
        char[] bucket = new char[26];
        char[] cs = s.toCharArray();
        for(int i=0;i<cs.length;i++){
            bucket[cs[i]-'a']++;
        }
        
        char[] ordercs = order.toCharArray();
        StringBuilder res = new StringBuilder();
        for(int i=0;i<ordercs.length;i++){
            while( bucket[ordercs[i]-'a']>0){
                res.append(ordercs[i]);
                 bucket[ordercs[i]-'a']--;
            }
        }
        for(int i=0;i<bucket.length;i++){
            while(bucket[i]>0){
                res.append((char)(i+'a'));
                bucket[i]--;
            }
        }
        return res.toString();
    }