TreeSet类

我们之前学习的类都无法对元素自动排序,java设置这个类能够将储存在集合中的元素自动排序,TreeSet类的底层使用的是二叉树结构。在储存对象时会根据对象的比较功能,确定对象的大小,从而确定该对象在二叉树中的位置。

import java.util.Iterator;
import java.util.TreeSet;

public class TreeSetDemo {

    public static void main(String[] args) {
        
        TreeSet ts=new TreeSet();
        ts.add("aaaa");
        ts.add("bbbb");
        ts.add("cccc");
        ts.add("dddd");
        for (Iterator iterator = ts.descendingIterator(); iterator.hasNext();) {
            System.out.println(iterator.next());
        } 
    }

}

底层TreeSet类在对对像进行排序时其实是将对象强转成cmparable,然后通过comparableTo方法进行比较大小,从而确定位置。
java中规定要想某个类具有比较功能,就要实现comparable接口。所以当TreeSet类存放自定义类时,该类要实现comparable接口。
用TreeSet类存放自定义对象:

import java.util.Iterator;
import java.util.TreeSet;

class People implements Comparable {

    private String name;

    private int age;

    public People(String name, int age) {
        super();
        this.name = name;
        this.age = age;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

    @Override
    public String toString() {
        return "People [name=" + name + ", age=" + age + "]";
    }

    @Override
    public int compareTo(Object o) {
        // TODO Auto-generated method stub
        People p = (People) o;
        return this.age - p.age;
    }

}

public class TreeSetDemo {
    public static void main(String[] args) {

        TreeSet ts = new TreeSet();
        ts.add(new People("aaaa", 128));
        ts.add(new People("m", 122));
        ts.add(new People("aaaa", 122));
        ts.add(new People("aana", 11));
        for (Iterator iterator = ts.descendingIterator(); iterator.hasNext();) {
            System.out.println(iterator.next());
        }
    }

}

你可能感兴趣的:(TreeSet类)