使用自定义字符顺序对Java字符串列表进行排序

import java.util.Comparator;

class MyComparator implements Comparator<String> {
    private String order; // 存储自定义字符顺序

    /**
     * 构造函数，初始化自定义字符顺序
     * @param order 定义字符优先级的字符串
     */
    MyComparator(String order) {
        this.order = order;
    }

    /**
     * 比较两个字符串的顺序
     * 该实现基于字符串的首字母在自定义顺序字符串中的位置进行比较。
     *
     * @param s1 第一个要比较的字符串
     * @param s2 第二个要比较的字符串
     * @return 如果s1的首字母在order中排在s2的首字母之前，返回负数；
     *         如果s1的首字母在order中排在s2的首字母之后，返回正数；
     *         如果首字母相同，返回0。
     */
    @Override
    public int compare(String s1, String s2) {
        // 获取两个字符串的首字母
        char charS1 = s1.charAt(0);
        char charS2 = s2.charAt(0);

        // 获取它们在自定义顺序字符串中的索引
        // indexOf()方法返回字符第一次出现的索引，如果不存在则返回-1
        int indexS1 = order.indexOf(charS1);
        int indexS2 = order.indexOf(charS2);

        // 通过索引的差值来决定排序
        return indexS1 - indexS2;
    }
}

import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

public class CustomStringSortExample {
    public static void main(String[] args) {
        // 示例数据
        List<String> names = Arrays.asList("Robin", "Zax", "John");
        String customOrder = "ZJR"; // 自定义字符顺序：Z优先，J次之，R再次之

        System.out.println("原始列表: " + names);

        // 使用Stream API和自定义Comparator进行排序
        List<String> sortedNames = names.stream()
                                        .sorted(new MyComparator(customOrder))
                                        .collect(Collectors.toList());

        System.out.println("排序结果: " + sortedNames);

        // 另一个示例
        List<String> names2 = Arrays.asList("Robin", "Zax", "John", "Rohan");
        String customOrder2 = "OZJRHBAS";
        System.out.println("\n原始列表2: " + names2);
        List<String> sortedNames2 = names2.stream()
                                         .sorted(new MyComparator(customOrder2))
                                         .collect(Collectors.toList());
        System.out.println("排序结果2: " + sortedNames2);
    }
}

原始列表: [Robin, Zax, John]
排序结果: [Zax, John, Robin]

原始列表2: [Robin, Zax, John, Rohan]
排序结果2: [Zax, John, Rohan, Robin]