Java基础必学TreeSet集合

public class TreeSetDemo {

    public static void main(String[] args) {

        //创建对象

        TreeSet<Integer>ts= new TreeSet<Integer>(); //注意：泛型中引用类型要用它的包装类类型

        //添加数据

        ts.add( 10 ); //自动装箱实现

        ts.add( 8 );

        ts.add( 7 );

        ts.add( 5 );

        ts.add( 10 ); //由于是Set集合不包含重复元素，所以该元素不显示

        //遍历方法一 增强for

        for (Integer i:ts){

            System.out.println(i);

        }

        System.out.println( "--------------------------------" );

        //遍历方法二 迭代器

        Iterator<Integer>it =ts.iterator();

        while (it.hasNext()){

            System.out.println(it.next());

        }

    }

}

package com.wkcto.chapter05.set;

import java.util.Comparator;

import java.util.TreeSet;

/**

  * TreeSet集合

  *      可以对元素进行自然排序, 要求元素必须是可比较的

  *          1)创建TreeSet集合时,通过构造方法指定Comparator比较器

  *          2)如果没有指定Comparator比较器, 要求元素的类必须实现Comparable接口

  * @author 蛙课网

  *

  */

public class Test03 {

     public static void main(String[] args) {

         //1) 创建TreeSet集合,存储Person对象, 在构造方法中指定Comparator比较器,按姓名升序排序

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

             //在匿名内部类中重写接口的抽象方法

             @Override

             public int compare(Person o1, Person o2) {

                 //指定一个比较规则

                 return o1.name测试数据pareTo(o2.name);   //姓名升序

             }

         });

         //2)添加元素

         treeSet.add( new Person( "lisi" , 18 ));

         treeSet.add( new Person( "feifei" , 28 ));

         treeSet.add( new Person( "yong" , 35 ));

         treeSet.add( new Person( "bin" , 36 ));

         treeSet.add( new Person( "zhang" , 33 ));

         //3)直接打印

         System.out.println( treeSet );

         //4) 根据已有TreeSet创建新的TreeSet集合

         TreeSet<Person> treeSet22 = new TreeSet<>(treeSet);

         System.out.println( treeSet22 );

         //5) 使用TreeSet的无参构造, 没有指定Comparator比较器, 要求Person类实现Comparable接口

         TreeSet<Person> treeSet33 = new TreeSet<>();

         treeSet33.addAll(treeSet);     

         System.out.println( treeSet33);

     }

}

package com.wkcto.chapter05.set;

import java.util.Comparator;

import java.util.TreeSet;

/**

  * 注意:

  *      在TreeSet集合中, 是根据Comparator/Comparable的比较结果是否为0来判断是否为同一个对象

  *      如果Comparator的compare()方法/Comparable的compareTo()方法的返回值为0 就认为是同一个对象

  * @author 蛙课网

  *

  */

public class Test04 {

     public static void main(String[] args) {

         //创建TreeSet集合,存储Person对象, 通过构造方法指定Comparator比较器,按年龄降序排序

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

             @Override

             public int compare(Person o1, Person o2) {

                 return o2.age - o1.age;

             }

         });

         //当前treeSet是根据年龄比较Person大小的, 在添加Person对象时, 如果年龄相同就认为是同一个对象

         treeSet.add( new Person( "lisi" , 18 ));

         treeSet.add( new Person( "feifei" , 18 ));

         treeSet.add( new Person( "zhang" , 18 ));

         treeSet.add( new Person( "yong" , 18 ));

         System.out.println( treeSet.size() );       // 1

         System.out.println( treeSet );

         System.out.println( treeSet.contains( new Person( "wang" , 18 )));   //true

     }

}