如何让Java的TreeSet/HashMap按插入顺序索引，无需LinkedHashMap？

嘿，这个问题问得挺有针对性的！咱们分开聊聊HashMap和TreeSet的情况，因为这俩的底层逻辑完全不一样：

先看HashMap：完全行不通 #

HashMap的底层是哈希表结构，它的存储顺序只和元素的hashCode()以及数组索引的计算逻辑有关，和compareTo()根本不沾边——毕竟HashMap从来没要求元素必须实现Comparable接口。不管你怎么修改compareTo()方法，HashMap都不可能按照插入顺序来存储元素。这种场景下，LinkedHashMap才是标准解决方案，它就是在HashMap的基础上额外维护了一个链表，专门用来记录插入顺序。

再看TreeSet：理论上可行，但有一堆坑 #

TreeSet是基于TreeMap实现的，它的排序规则完全依赖元素的compareTo()方法（或者自定义的Comparator）。要让它按插入顺序排列，核心思路是给每个元素绑定一个唯一的插入序列号，然后让compareTo()比较这个序列号，而不是元素本身的值。

举个实际的代码例子： #

class InsertionOrderedElement implements Comparable<InsertionOrderedElement> {
    // 全局自增计数器，给每个新元素分配唯一的插入序号
    private static int insertionCounter = 0;
    private final int value;
    // 每个元素的插入顺序标记，一旦创建就不可修改
    private final int insertionOrder;

    public InsertionOrderedElement(int value) {
        this.value = value;
        this.insertionOrder = insertionCounter++;
    }

    public int getValue() {
        return value;
    }

    @Override
    public int compareTo(InsertionOrderedElement other) {
        // 按插入序号排序，先插入的元素序号小，排在前面
        return Integer.compare(this.insertionOrder, other.insertionOrder);
    }
}

测试一下这段代码：

public class TestTreeSet {
    public static void main(String[] args) {
        TreeSet<InsertionOrderedElement> set = new TreeSet<>();
        set.add(new InsertionOrderedElement(1));
        set.add(new InsertionOrderedElement(19));
        set.add(new InsertionOrderedElement(3));
        set.add(new InsertionOrderedElement(4));
        set.add(new InsertionOrderedElement(2));
        set.add(new InsertionOrderedElement(0));

        // 遍历输出顺序就是插入顺序：1, 19, 3, 4, 2, 0
        set.forEach(elem -> System.out.println(elem.getValue()));
    }
}

但这里有几个致命的坑： #

1. 违背TreeSet的设计初衷：TreeSet本来是用来实现基于元素值排序的有序集合，现在强行改成按插入顺序排列，相当于用菜刀砍骨头——虽然能完成，但不是合适的工具。
2. 去重逻辑的冲突：TreeSet判断元素是否重复的标准是compareTo()返回0，而不是equals()或hashCode()。如果我们的compareTo()只比较插入序号，那两个值相同但插入顺序不同的元素，会被TreeSet当成不同元素存入，直接破坏了Set的去重特性。如果要兼顾去重（相同值只留一个），那compareTo()必须在值相同时返回0，但这样后插入的相同值元素会被拒绝，而且也没法维护插入顺序了。
3. 不可变要求：插入序号必须是不可变的，如果后续修改了这个序号，会直接破坏TreeSet内部的红黑树结构，导致集合混乱。

总结 #

• HashMap：别浪费时间了，完全不可能通过修改compareTo()实现插入顺序存储，直接用LinkedHashMap就好。
• TreeSet：理论上能实现，但会带来一堆逻辑冲突和设计矛盾，实际开发中完全没必要——直接用LinkedHashSet才是最优解，它既保留了HashSet的去重特性，又通过链表完美维护了插入顺序。

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