自定义字符串排序（java）

问题描述：
给定两个字符串 order 和 s 。order 的所有单词都是 唯一 的，并且以前按照一些自定义的顺序排序。
对 s 的字符进行置换，使其与排序的 order 相匹配。更具体地说，如果在 order 中的字符 x 出现字符 y 之前，那么在排列后的字符串中， x 也应该出现在 y 之前。
返回 满足这个性质的 s 的任意排列 。

样例如下：

代码如下（想法就是先将s中存在的order字符按照order中顺序输出，再将别的字符放到最后的位置）：

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;

public class CustomSortString {
    //给定两个字符串 order 和 s 。order 的所有单词都是 唯一 的，并且以前按照一些自定义的顺序排序。
    //对 s 的字符进行置换，使其与排序的 order 相匹配。更具体地说，如果在 order 中的字符 x 出现字符 y 之前，
    // 那么在排列后的字符串中， x 也应该出现在 y 之前。
    //返回 满足这个性质的 s 的任意排列 。
    public static String customSortString(String order, String s) {
        HashSet<Character> hashSet = new HashSet<>();
        for (int i = 0; i < order.length(); i++) {
            hashSet.add(order.charAt(i));
        }
        StringBuilder stringBuilder = new StringBuilder();//用于存储未在order中的字符
        Map<Character,Integer> map = new HashMap<>();//用于存储在s中包含order字符的个数
        for (int i = 0; i < s.length(); i++) {
            if (hashSet.contains(s.charAt(i))){
                if (map.containsKey(s.charAt(i))){
                    int temp = map.get(s.charAt(i));
                    map.put(s.charAt(i),++temp);
                }else {
                    map.put(s.charAt(i),1);
                }
            }else {
                stringBuilder.append(s.charAt(i));
            }
        }
        StringBuilder res = new StringBuilder();
        for (int i = 0; i < order.length(); i++) {
            if (map.containsKey(order.charAt(i))){
                int temp = map.get(order.charAt(i));
                for (int j = 0; j < temp; j++) {
                    res.append(order.charAt(i));
                }
            }
        }
        res.append(stringBuilder);
        return res.toString();
    }

    public static void main(String[] args) {
        System.out.println(customSortString("cba","abcd"));
        System.out.println(customSortString("cbafg","abcd"));

    }
}

结果如下：