TreeMap自定义排序时的歧义问题及代码输出解析

代码示例 #

public static void main(String[] args) {
    String[] s = {"abc", "def", "ghi"};
    TreeMap<String, Integer> tm = new TreeMap<>((a, b) -> -a.length() + b.length());

    for (String k : s) {
        tm.merge(k, 1, Integer::sum);
    }

    System.out.println(tm);
}

用户预期输出 #

{abc=1, def=1, ghi=1}

歧义点与解析 #

1. 自定义比较器的逻辑
   代码里的比较器(a, b) -> -a.length() + b.length()等价于b.length() - a.length()，作用是按字符串长度升序排序。但当两个字符串长度相同时，比较器会返回0。

2. TreeMap的核心判定规则
   TreeMap判断两个key是否为同一key的依据是比较器返回值是否为0，和equals()方法的结果无关。也就是说，只要比较器返回0，TreeMap就认定这两个key是同一个，不会新增条目，而是执行merge操作里的value合并逻辑（这里是Integer::sum，即累加value）。

3. 预期与实际的矛盾
   按照用户的预期，三个长度相同的字符串应该作为独立key存在，但实际运行时，TreeMap会把这三个key视为同一个，最终输出是{abc=3}（因为每次merge都在同一个key的value上累加1）。这种歧义来源于对TreeMap比较器作用的误解——它不仅负责排序，还直接决定了key的唯一性。

4. 解决歧义的方法
   如果希望长度相同的key能被独立存储，需要在比较器里补充长度相同时的排序规则，比如按字符串自然顺序排序：

   TreeMap<String, Integer> tm = new TreeMap<>((a, b) -> {
       int lenDiff = b.length() - a.length();
       return lenDiff != 0 ? lenDiff : a.compareTo(b);
   });
   

   这样，长度相同的key会通过自然顺序区分，merge时不会合并value，最终输出就会符合用户预期。

内容的提问来源于stack exchange，提问作者Vishant Sandilya