TreeSet的用法举例

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

/*
1、每个集合对象的创建
2、向集合中添加元素
3、从集合中取出某个元素
4、遍历集合
5、测试TreeSet集合中的元素是：可排序的
6、测试TreeSet集合中存储的类型是自定义的。
7、测试实现Comparable接口的方式
8、测试实现Comparator接口的方式（最好测试以下匿名内部类的方式)

 */
public class 总结TreeSet {
    public static void main(String[] args) {
        //创建对象
        //TreeSet<Integer> ts = new TreeSet<>();
        //编写比较器可以改变规则
        TreeSet<Integer> ts = new TreeSet<>(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2 - o1;
            }
        });
        //添加元素
        ts.add(1);
        ts.add(20);
        ts.add(13);
        ts.add(4);
        //set集合没有下标，所以无法单独取出某一个元素
        //遍历（迭代器方式）
        Iterator<Integer> it = ts.iterator();
        while(it.hasNext()){
            System.out.println(it.next());
        }
        //遍历(foreach方式)
        for(Integer i:ts){
            System.out.println(i);
        }
        //TreeSet集合中存储自定义类型
        //创建对象
        TreeSet<A> tss = new TreeSet<>();
        //添加元素
        tss.add(new A(100));
        tss.add(new A(200));
        tss.add(new A(1020));
        tss.add(new A(1400));
        tss.add(new A(2100));
        tss.add(new A(10));
        tss.add(new A(12));
        //遍历
        for(A a:tss){
            System.out.println(a);
        }
        //创建对象
      //TreeSet<B> tb = new TreeSet<>(new BComparator());//比较器的方式
        TreeSet<B> tb = new TreeSet<>(new Comparator<B>() {//匿名内部类的方式
            @Override
            public int compare(B o1, B o2) {
                return o1.i - o2.i;
            }
        });
        //加元素
        tb.add(new B(1));
        tb.add(new B(12));
        tb.add(new B(471));
        tb.add(new B(10));
        tb.add(new B(21));
        //遍历
        for(B b:tb){
            System.out.println(b);
        }
    }
}
//第一种方式：实现Comparable接口
class A implements Comparable<A>{
    int i;

    public A(int i) {
        this.i = i;
    }

    @Override
    public String toString() {
        return "A{" +
                "i=" + i +
                '}';
    }

    @Override
    public int compareTo(A o) {
        return this.i-o.i;
    }
}
//第二种方式：
class B{
    int i;

    public B(int i) {
        this.i = i;
    }

    @Override
    public String toString() {
        return "B{" +
                "i=" + i +
                '}';
    }
}
//比较器
class BComparator implements Comparator<B>{

    @Override
    public int compare(B o1, B o2) {
        return o1.i - o2.i;
    }
}