关于TreeSet的排序对于删除操作的影响

先贴上准备的代码：

TreeSet<Node> list =newTreeSet<>();
Node n1 =newNode(1);
Node n2 =newNode(2);
Node n3 =newNode(3);
Node n4 =newNode(4);
Node n5 =newNode(5);

这是Node，继承Comparable接口

protected static class Node implements Comparable<Node>
{
public int id;
private int __id;
private static int ID =1;
public Node(int id){
this.id = id;
this.__id = ID++;
}
@Override
public String toString(){
return"Node("+__id+") "+id;
}
@Override
public int compareTo(Node o){
if(o.id>id)
return ;
else if(o.id == id)
return 0;
else
return -1;
}
}

TreeSet是一个二叉树集合，对于TreeMap的一个封装，增加一个元素，其实就是把元素当成Key，固定一个value放入TreeMap。

如果我们要研究TreeSet的排序，那么不得不了解TreeMap的Entry，其实就是一个二叉树结构，对于TreeSet本身来说，功能都是建立在TreeMap的基础上。

TreeMap$Entry拥有除了Key和Value之外，多了三个属性，left，right，parent。left和right作为二叉树的分叉，parent建立树的上下结构。

插入一个元素，都是根据比较器Compare出大小，然后从根节点开始，建立二叉树。所以在插入和删除的时候，有一个重要的地方，就是当你需要修改某一个元素的时候，必须确保这个元素的排序是没有被影响过的。比如我们上面一个列子，1-5个元素插入之后，当需要删除元素1，就必须保证，Node的id为1，如果id被修改成非1的情况下，再通过remove(Object obj)来删除就不能找到节点，因此删除就不可能了。

上测试例子：

package examples.base;

import java.util.TreeSet;

public class TestTreeSet {
    public static void main(String[] args) {
        TreeSet<Node> list = new TreeSet<>();
        Node n1 = new Node(1);
        Node n2 = new Node(2);
        Node n3 = new Node(3);
        Node n4 = new Node(4);
        Node n5 = new Node(5);
        Node n = new Node(5);
        list.add(n1);
        list.add(n2);
        list.add(n3);
        list.add(n4);
        list.add(n5);
        list.add(n);
        System.out.println(list);
        System.out.println(list.higher(n3));
        n1.id = 5;
        if(list.remove(n1)){
            System.out.println("true");
        }
        
        list.add(n1);
        System.out.println(list);
        
    }
    
    protected static class Node implements Comparable<Node>
    {
        public int id;
        private int __id;
        private static int ID = 1;
        public Node(int id) {
            this.id = id;
            this.__id = ID++;
        }
        
        @Override
        public String toString() {
            return "Node("+__id+") "+id;
        }
        
        @Override
        public int compareTo(Node o) {
            if(o.id>id)
                return 1;
            else if(o.id == id)
                return 0;
            else
                return -1;
                        
        }
    }
}

结果：

[Node(5)5,Node(4)4,Node(3)3,Node(2)2,Node(1)1]
Node(2)2
true
[Node(1)5,Node(4)4,Node(3)3,Node(2)2,Node(1)5]

删除成功其实只是删除了元素5，而元素1虽然id是5，但是它还是老的排序，所以在二叉树中是定位不到的，而因为id为5的元素存在，而且排序没有被影响，因此删除就能够成功。

最后，我们谨记一个要点：

当我们用TreeSet作为一个自动排序队列的时候，更新元素的位置，必须分三个步骤：

1、remove老的元素

2、修改

3、插入修改后的元素

来自为知笔记(Wiz)

秒客网

关于TreeSet的排序对于删除操作的影响

相关文章