Set容器——TreeSet及常用API

TreeSet及常用Api

①   TreeSet为使用树来进行存储的Set接口提供了一个工具，对象按升序存储，访问和检索很快;

②   在存储了大量的需要进行快速检索的排序信息的情况下，TreeSet是一个很好的选择;

③   构造方法：

a)   TreeSet();

b)   TreeSet(Collection c)

c)    TreeSet(Comparator comp)

d)    TreeSet(SortedSet ss)

总结：TreeSet的内部操作的底层数据是TreeMap，只是我们操作的是TreeMap的key;

创建一个Person类：

class Person{
    private String name;
    private int age;
    public Person(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;
    }
}

主方法中创建实例，添加元素，输出：

        TreeSet<Person> pset=new TreeSet<Person>();
        pset.add(new Person("张三",20));
        pset.add(new Person("李四",30));
        pset.add(new Person("王五",40));
        pset.add(new Person("小明",50));
        System.out.println(pset);

输出结果：

报错，与TreeMap相同没有比较方法

解决方法与TreeMap中的一样，

方法一：实现Comparable接口的方法

 class Person implements Comparable<Person>{ 2 } 

重构方法如下

    @Override
    public int compareTo(Person o) {
        if (this.age - o.getAge() > 0) {
            return 1;
        } else if (this.age - o.getAge() < 0) {
            return -1;
        }
        return 0;
    }

再在主方法中使用迭代器输出TreeMap中的值

        Iterator<Person> it = pset.iterator();
        while(it.hasNext()){
            Person p=it.next();
            System.out.println(p.getName()+"--"+p.getAge());
        }

输出结果：

张三--20

李四--30

王五--40

小明—50

 

方法二：使用匿名内部类

    TreeSet<Person> pset = new TreeSet<Person>(new Comparator<Person>() {

            @Override
            public int compare(Person o1, Person o2) {
                if(o1.getAge()>o2.getAge()){
                    return 1;
                }
                else if(o1.getAge()<o2.getAge()){
                    return -1;
                }
                return 0;
            }
        });

输出结果与上面相同