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>{ }

重构方法如下

     @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;
             }
         });

输出结果与上面相同