如何基于自定义比较器对键上的哈希图进行排序

时间:2019-07-12 19:21:15

标签: sorting collections java-8 custom-compare

我收到了一个常见的哈希表,需要使用自定义比较器对Key进行排序。

下面是我尝试过的方法,但是它似乎不起作用-它没有对键进行排序。

地图的键的格式为long-string-short 例如:( 11169-SW-1 / 11169-SW-2 / 11132-SH-2 / 11132-SH-7 / 11132-SH-1) 。字符串比较不起作用,因为我需要数字部分,因此我定义了一个自定义比较器。我意识到自定义比较器代码需要清理-例如需要合并多个return语句,并且还需要进行清理,但是一旦它起作用,我就可以这样做。

我该怎么办?

Map<String, Map<String, String>> strValuesMap = Utilities.getDataMapFromDB();

Map<String, Map<String, String>> sortedMap = new TreeMap<>(new CustomComparator());
sortedMap.putAll(strValuesMap);
sortedMap.forEach((x, y) -> System.out.println("id " + x + "=" + y));

自定义比较器的定义如下

class CustomComparator implements Comparator<String> {

    @Override
    public int compare(String plId1, String plId2) {

        System.out.println("plId1 : " + plId1 + " plId2 " + plId2);
        String[] plId1Split = plId1.split("-");
        String[] plId2Split = plId2.split("-");
        int retValue = 0;
        if (!plId1Split[0].equalsIgnoreCase(plId2Split[0])) {
            Long seq1 = new Long(plId1Split[0]);
            Long seq2 = new Long(plId2Split[0]);
            retValue = seq1.compareTo(seq1);
        }
        if (retValue != 0) {
            return retValue;
        }
        if (!plId1Split[1].equalsIgnoreCase(plId2Split[1])) {
            retValue = plId1.compareTo(plId2);
        }
        if (retValue != 0) {
            return retValue;
        } else {
            Short seq1 = new Short(plId1Split[2]);
            Short seq2 = new Short(plId2Split[2]);
            retValue = seq1.compareTo(seq2);
            return retValue;
        }
    }
}

谢谢

1 个答案:

答案 0 :(得分:1)

这是不正确的:

if (!plId1Split[1].equalsIgnoreCase(plId2Split[1])) {
    retValue = plId1.compareTo(plId2); // This part is wrong
}

在条件主体中,您应该只比较字符串的第二部分(plId1Split[1]plId2Split[1]),而比较整个字符串(plId1plId2) 。
因此,这意味着字符串的最后部分(plId1Split[2]plId2Split[2])不是按照数字顺序而是按照字典顺序进行比较。

应该是:

if (!plId1Split[1].equalsIgnoreCase(plId2Split[1])) {
    retValue = plId1Split[1].compareTo(plId2Split[1]);
}

关于比较器的清晰度,我认为您不需要进行某些测试。
例如,您与equalsIgnoreCase()进行比较,然后与long比较器进行比较,然后将同一字符串转换为long:

if (!plId1Split[0].equalsIgnoreCase(plId2Split[0])) {
        Long seq1 = new Long(plId1Split[0]);
        Long seq2 = new Long(plId2Split[0]);
        retValue = seq1.compareTo(seq1);
} 

总的来说,如果调用这两个比较,这并不是一个很好的优化,而且还会使代码的可读性降低。

还请注意,您过度使用了一些数字对象,这会产生不希望有的装箱操作(Long-> long),这会产生一定的成本。
您可以这样写:

class CustomComparator implements Comparator<String> {

    @Override
    public int compare(String plId1, String plId2) {

        String[] plId1Split = plId1.split("-");
        String[] plId2Split = plId2.split("-");

        int retValue = Long.compare(Long.parseLong(plId1Split[0]), 
                               Long.parseLong(plId2Split[0]));

        if (retValue != 0) {
            return retValue;
        }

        retValue = plId1Split[1].compareTo(plId2Split[1]);

        if (retValue != 0) {
            return retValue;
        }

        return Short.compare(Short.parseShort(plId1Split[2]), 
                       Short.parseShort(plId2Split[2]));

   }                    
}

作为替代方案,您可以依靠代表对象的自定义类来使用Java 8比较器:

// private package if makes sense to make this class not visible outside that
class Identifier {
    private long part1;
    private String part2;
    private short part3;

    Identifier(String[] split) {
        this.part1 =  Long.parseLong(split[0]);
        this.part2 =  split[1];
        this.part3 =  Short.parseShort(split[2]);
    }

    long getPart1() {
        return part1;
    }    
    String getPart2() {
        return part2;
    }
    short getPart3() {
        return part3;
    }
}

并使用public static <T, U> Comparator<T> comparing( Function<? super T, ? extends U> keyExtractor, Comparator<? super U> keyComparator)

提取排序键以比较元素,并为此排序键应用指定的比较器:

import static java.util.Comparator.comparing;
import static java.util.Comparator.comparingLong;

Comparator<String> comp =
        comparing(s -> new Identifier(s.split("-")),
                  comparingLong(Identifier::getPart1)
                 .thenComparing(Identifier::getPart2)
                 .thenComparingInt(Identifier::getPart3));

它清楚地揭示了比较/功能逻辑。

请注意,由于Java不提供元组的内置结构,因此我们需要引入自定义类来保存数据。
使用元组类(来自https://www.javatuples.org/或任何源代码),代码将更加简单。