自定义 Comparator 错误导致的treemap 取值为null

最新推荐文章于 2025-03-24 17:13:23 发布

LVZHULIU

最新推荐文章于 2025-03-24 17:13:23 发布

阅读量845

点赞数

CC 4.0 BY-SA版权

文章标签： java Comparator TreeMap

本文链接：https://blog.youkuaiyun.com/LVZHULIU/article/details/86585108

在实现一个需求时，使用TreeMap进行倒序遍历K-V数据，通过自定义Comparator遇到了NullPointerException。问题在于Comparator在判断相等时使用了`==`，而不是equals方法，导致包装类对象不恒等。此外，对TreeMap的内部原理理解不足也是原因。当插入重复key时，由于Comparator的bug，实际存储结构变为4.3<--4.2<--4.1，遍历时查找4.1的下一个节点为空，引发NP。解决方法是修正Comparator的比较逻辑。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近再做一个需求，kv的数据格式，要求对 k进行倒序遍历，于是想到了TreeMap，因为 TreeMap 默认升序排列，因此需要自定义一个Comparator，代码如下：

public static Comparator<Double> comparator = new Comparator<Double>() {

		@Override
		public int compare(Double o1, Double o2) {
			// TODO Auto-generated method stub
			if (o1 < o2) {
				return 1;
			} else if (o1 == o2) {//错误的地方
				return 0;
			} else {
				return -1;
			}
		}

	};

TreeMap 代码段：

Map<Double, Double> negativeMap = new TreeMap<Double, Double>(Constant.comparator);
positiveMap.put(4, 4);
positiveMap.put(4, 4);
positiveMap.put(4, 4);
double i=1;
for(Double key: negativeMap.keySet()){
		Double tmpValue = negativeMap.get(key);
		result += tmpValue/i;//出现空指针
		i++;
}

这段代码正常看无论都不会出现空指针，我正常put三条数据，即使key 相等，也会覆盖掉，不会出现NP，但事实就是出现了，而且打印的map size 是3，其中的问题就是：

1、Comparator 在判断相等条件时候用的 ==，而判断的对象为包装类，即使两个值相等，对象也不恒等

2、对 TreeMap的内部原理了解不够

以上面的例子说明：