Java TreeMap使用自定义对象作为键的类型转换异常解决

问题原因 #

TreeMap是有序映射，默认依赖键的「自然排序规则」工作——要求键必须实现java.lang.Comparable接口，否则在插入键值对时，JVM会尝试将键强制转换为Comparable类型，从而抛出class RecyclingJava.recycling.knot cannot be cast to class java.lang.Comparable异常。

而HashMap不需要排序，仅依赖哈希值定位元素，所以之前用HashMap时不会出现这个问题。

两种解决方案 #

方案一：让knot类实现Comparable接口 #

修改knot类的定义，实现Comparable<knot>并重写compareTo方法，定义对象间的比较规则。如果需要每个knot实例作为唯一键，可基于对象的身份哈希值实现比较：

package RecyclingJava.recycling;

import java.util.Map;
import java.util.TreeMap;

public class knot implements Comparable<knot> {

    Map<knot, Boolean> connect = new TreeMap<>(); //TreeMap

    @Deprecated
    @Override
    protected void finalize() throws Throwable {
        System.out.println("Finalize object");
        super.finalize();
    }

    public void addconnection(knot knot) {
        this.connect.put(knot, true);
    }

    // 重写compareTo方法，定义排序规则
    @Override
    public int compareTo(knot other) {
        // 基于对象身份哈希值比较，保证每个实例唯一区分
        return Integer.compare(System.identityHashCode(this), System.identityHashCode(other));
    }

    // 建议同时重写equals和hashCode，保证逻辑一致性
    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        knot knot = (knot) o;
        return System.identityHashCode(this) == System.identityHashCode(knot);
    }

    @Override
    public int hashCode() {
        return System.identityHashCode(this);
    }

    public static void main(String[] args) {
        for (int i = 0; i <= 1000000; i++) {
            knot w1 = new knot();
            knot w2 = new knot();
            w1.addconnection(w2);
            w2.addconnection(w1);
        }
    }
}

方案二：初始化TreeMap时传入自定义Comparator #

无需修改knot类，直接在创建TreeMap实例时，传入一个Comparator<knot>实现，指定键的比较规则：

package RecyclingJava.recycling;

import java.util.Map;
import java.util.TreeMap;

public class knot {

    // 初始化TreeMap时传入自定义比较器
    Map<knot, Boolean> connect = new TreeMap<>((k1, k2) -> 
        Integer.compare(System.identityHashCode(k1), System.identityHashCode(k2))
    );

    @Deprecated
    @Override
    protected void finalize() throws Throwable {
        System.out.println("Finalize object");
        super.finalize();
    }

    public void addconnection(knot knot) {
        this.connect.put(knot, true);
    }

    public static void main(String[] args) {
        for (int i = 0; i <= 1000000; i++) {
            knot w1 = new knot();
            knot w2 = new knot();
            w1.addconnection(w2);
            w2.addconnection(w1);
        }
    }
}

注意事项 #

• 如果业务需要基于knot类的某个属性（比如自定义ID）排序，只需把compareTo或Comparator中的逻辑替换为对应属性的比较即可。
• 无论用哪种方案，都建议重写equals和hashCode方法，保证对象相等性判断与排序/哈希逻辑一致，避免出现潜在的逻辑错误。

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