自定义字符串排序

/*
题目思路:
1. 使用 HashMap 存储 order 中字符的顺序。
2. 将 s 字符串转换为流，按顺序和不按顺序分类。
3. 按照 order 中的顺序排序后与未排序部分结合。
*/
import java.util.*;
import java.util.stream.*;

public class CustomSortString {
    public String customSortString(String order, String s) {
        // 存储 order 中字符的顺序
        Map<Character, Integer> orderMap = new HashMap<>();
        for (int i = 0; i < order.length(); i++) {
            orderMap.put(order.charAt(i), i);
        }
        
        // 使用 Java Stream API 处理字符分类和排序
        String result = s.chars()
            .mapToObj(c -> (char) c)
            .sorted(Comparator.comparingInt(c -> orderMap.getOrDefault(c, Integer.MAX_VALUE)))
            .map(String::valueOf)
            .collect(Collectors.joining());
        
        return result;
    }
}