java - 使用自定义比较器在 Java 中创建 SortedMap

Question

我想在 Java 中创建一个具有自定义排序顺序的 TreeMap。排序后的键是字符串，需要根据第二个字符进行排序。值也是字符串。

示例 map :

Za,FOO
Ab,Bar

Answer 1

您可以像这样使用自定义比较器:

    Comparator<String> secondCharComparator = new Comparator<String>() {
        @Override public int compare(String s1, String s2) {
            return s1.substring(1, 2).compareTo(s2.substring(1, 2));
        }           
    };

示例:

    SortedMap<String,String> map =
        new TreeMap<String,String>(secondCharComparator);
    map.put("Za", "FOO");
    map.put("Ab", "BAR");
    map.put("00", "ZERO");
    System.out.println(map); // prints "{00=ZERO, Za=FOO, Ab=BAR}"

请注意，这只是假设 String 在索引 1 处有一个字符。如果没有，它会抛出 StringIndexOutOfBoundsException。

---

或者，您也可以使用这种比较:

return s1.charAt(1) - s2.charAt(1);

这个减法“技巧”通常被破坏，但它在这里工作正常，因为两个 char 的减法不会溢出 int。

不过，上面的 substring 和 compareTo 解决方案更具可读性。

另见:

• Java Integer: what is faster comparison or subtraction?