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.compareTo(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
	}

}