TreeSet底层实际是用TreeMap实现的,内部维持了一个简化版的TreeMap,通过key来存储Set的元素。 TreeSet内部需要对存储的元素进行排序,因此,我们对应的类需要实现Comparable接口。这样,才能根据compareTo()方法比较对象之间的大小,才能进行内部排序。
package cn.sxt.mycollection;
import java.util.Set;
import java.util.TreeSet;
/**
*
- 测试TreeSet使用,Comparable接口(与TreeMap基本相同)
- @author 骚彬
*/
public class TestTreeSet {
public static void main(String[] args) {
Set set=new TreeSet<>();
set.add(300);
set.add(200);
set.add(600);
//*元素递增排序
for(Integer m:set) {
System.out.println(m);
}
Set<Emp2> set2=new TreeSet<>();
set2.add(new Emp2(100,"mb",3000));
set2.add(new Emp2(50,"b",2000));
set2.add(new Emp2(150,"m",8000));
set2.add(new Emp2(30,"bm",20000));
for(Emp2 m:set2) {
System.out.println(m);
}
}
}
class Emp2 implements Comparable{ ////接口
int id;
String name;
double salary;
public Emp2(int id, String name, double salary) {
super();
this.id = id;
this.name = name;
this.salary = salary;
}
@Override
public String toString() {
return “id:”+id+",name:"+name+",salary:"+salary;
}
@Override
public int compareTo(Emp2 o) { //负数:小于 0:等于,正数:大于
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<o.id) {
return -1;
}
return 0;
}
}
}
使用TreeSet要点:
(1) 由于是二叉树,需要对元素做内部排序。 如果要放入TreeSet中的类没有实现Comparable接口,则会抛出异常:java.lang.ClassCastException。
(2) TreeSet中不能放入null元素。