TreeSet的使用和底层原理

TreeSet的底层

TreeSet底层实际是用TreeMap实现的，内部维持了一个简化版的TreeMap，通过key来存储Set的元素。 TreeSet内部需要对存储的元素进行排序，因此，我们对应的类需要实现Comparable接口。这样，才能根据compareTo()方法比较对象之间的大小，才能进行内部排序。

TreeSet和Comparable接口的使用

Comparable接口的实现

class Emp implements Comparable<Emp>{
	int id;
	String name;
	int salary;
	public Emp(int id, String name, int salary) {
		super();
		this.id = id;
		this.name = name;
		this.salary = salary;
	}
	@Override
	public int compareTo(Emp o) {
		if(this.salary>o.salary) {
			return -1;
		}else if(this.salary<o.salary) {
			return 1;
		}else {
			if(this.id>o.id) {
				return 1;
			}else if(this.id<id){
				return -1;
			}
		}
		return 0;
	}
	public String toString() {
		return "id:"+id+"\t"+"name:"+name+"\t"+"salary:"+salary;
	}

TreeSet的使用

public class TestTreeSet {
	public static void main(String[] args) {
		Set<Integer> set = new TreeSet<>();
		//按照元素递增方式排好序
		set.add(500);
		set.add(60);
		set.add(99);
		for(Integer m:set) {
			System.out.println(m);
		}
		Emp e1 = new Emp(1001,"阿狸",666666);
		Emp e2 = new Emp(99,"阿卡丽",666668);
		Emp e3 = new Emp(120,"瑞雯",527788);
		Set<Emp> set2 = new TreeSet<>();
		set2.add(e1);
		set2.add(e2);
		set2.add(e3);
		for(Emp e:set2) {
			System.out.println(e);
		}
	}
}

效果

使用TreeSet的要点

(1) 由于是二叉树，需要对元素做内部排序。 如果要放入TreeSet中的类没有实现Comparable接口，则会抛出异常：java.lang.ClassCastException。

(2) TreeSet中不能放入null元素。