Java 自定义比较器实现

Question

我正在编写一个比较器，我可以使用它使用字符数对字符串数组进行排序。 字符串都是小写的 ASCII 字符。

例如。

input = {"art", "bash", "tar"};
Expected sorted output = {"art", "tar", "bash"}; 

在这里，“bash”是否在开头并不重要，但“art”和“tar”应该一个接一个，没有特定的顺序，因为它们具有匹配的字符数。

我有这段代码，我在其中使用一个简单的数组来检查和保持字符数并进行比较。

Comparator<String> comparator = new Comparator<String>() {
    
    @Override
    public int compare(String s1, String s2){
        if (s1.equals(s2)) return 0;
        
        int[] a1 = getCountArray(s1);
        int[] a2 = getCountArray(s2);
        if (Arrays.equals(a1, a2)) return 0; // character count matches

        return s1.compareTo(s2);
    }

    private int[] getCountArray(String s) {
        int[] array = new int[256];
        for (int i=0; i<s.length(); i++){
            array[s.charAt(i) - 'a'] ++;
        }
        return array;
    }

};
String[] input = {"art", "bash", "tar"};
Arrays.sort(input, comparator); 

但是，它不起作用。 我究竟做错了什么？

Answer 1

您的比较器不一致：

• 当比较“art”和“bash”时，它返回“art”小于“bash”
• 当比较“bash”和“tar”时，它返回“bash”小于“tar”
• 因此，当比较“art”和“tar”时，它必须返回“art”小于“tar”（因为“art”<“bash”<“tar”），但比较器返回的“art”与“tar”相同”。

当您调用Arrays.sort(input, comparator); sort 方法将进行成对比较，并且总是会发现“bash”必须在“tar”之前。

要解决此问题，您必须将return s1.compareTo(s2)替换为比较字符数 arrays a1和a2的内容。