Java TreeMap put vs HashMap put，自定义对象作为键

Question

我的目标是使用 TreeMap 使 Box 键对象按 Box.volume 属性排序，同时能够将键按 Box.code 区分。在 TreeMap 中不可以吗？

根据下面的测试 1，HashMap put 按预期工作，HashMap 保留 A、B 键对象，但在测试 2 中，TreeMap put 不将 D 视为不同的键，它替换C 的值，请注意，我将 TreeMap 比较器用作 Box.volume，因为我希望键在 TreeMap 中按体积排序。

import java.util.*;

public class MapExample {
    public static void main(String[] args) {
        //test 1
        Box b1 = new Box("A");
        Box b2 = new Box("B");
        Map<Box, String> hashMap = new HashMap<>();
        hashMap.put(b1, "test1");
        hashMap.put(b2, "test2");
        hashMap.entrySet().stream().forEach(o-> System.out.println(o.getKey().code+":"+o.getValue()));
        //output
        A:test1
        B:test2

        //test 2
        Box b3 = new Box("C");
        Box b4 = new Box("D");
        Map<Box, String> treeMap = new TreeMap<>((a,b)-> Integer.compare(a.volume, b.volume));
        treeMap.put(b3, "test3");
        treeMap.put(b4, "test4");
        treeMap.entrySet().stream().forEach(o-> System.out.println(o.getKey().code+":"+o.getValue()));
        //output
        C:test4
    }
}

class Box {
    String code;
    int volume;

    public Box(String code) {
        this.code = code;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        Box box = (Box) o;
        return code.equals(box.code);
    }

    @Override
    public int hashCode() {
        return Objects.hash(code);
    }
}

谢谢

Answer 1

TreeMap 将比较方法返回 0 的 2 个键视为相同，即使它们彼此不相等，因此您当前的 TreeMap 不能包含两个键具有相同的体积。

如果你想保持按体积排序，并且在你的Map中仍然有多个具有相同体积的键，改变你的Comparator的比较方法来比较Box 体积相等时的代码。这样它只会在键相等时返回 0。

Map<Box, String> treeMap = new TreeMap<>((a,b)-> a.volume != b.volume ? Integer.compare(a.volume, b.volume) : a.code.compareTo(b.code));

现在的输出是:

C:test3
D:test4