TreeSet的使用

TreeSet的存储结构及排序方法

●存储结构:红黑树

---

●基于排列顺序实现元素不重复。
●实现了SortedSet接口，对集合元素自动排序。
●元素对象的类型必须实现Comparable接口， 指定排序规则。
●通过CompareTo方法确定是否为重复元素。

添加元素

        //对象的类型必须实现Comparable接口,然后重写一个CompareTo()方法
        TreeSet<Person> person=new TreeSet<>();
        Person p1=new Person("ldh",20);
        Person p2=new Person("lzl",21);
        Person p3=new Person("lcw",26);
        Person p4=new Person("zxc",24);
        person.add(p1);
        person.add(p2);
        person.add(p3);
        person.add(p4);
        System.out.println("集合元素个数为："+person.size());
        System.out.println(person.toString());

删除元素

        person.remove(p1);
        person.remove(new Person("zxc",24));
        System.out.println("删除后的元素个数为："+person.size());

遍历元素

        //增强for循环
        for (Person person1 : person) {
            System.out.println(person1);
        }
        System.out.println("--------------------");
        //迭代器
        Iterator it=person.iterator();
        while (it.hasNext()){
            System.out.println(it.next());
        }

判断

        System.out.println(person.contains(p1));
        System.out.println(person.contains(new Person("zxc",24)));
        System.out.println(person.isEmpty());

此文中所用到的重写方法

需要先将Person类实现Comparable接口,然后重写一个CompareTo()方法

    @Override
    public int compareTo(Person o){
        int n1=this.getName().compareTo(o.getName());
        int n2=this.age-o.getAge();
        return n1==0?n2:n1;
    }