TreeSet使用元素的自然顺序(暂时理解为unicode顺序)对集合中的元素进行排序。添加的元素必须自己实现Comparable接口,以便排序时默认(源码理解)调用其compareTo方法进行比较,TreeSet本身不须实现Comparable接口。
为保证元素唯一性,当有相同元素时,不能往里添加。判断元素唯一性的方式:根据比较方法返回的结果是不是0(public int compareTo(Object o)
)。
//由于String类已经实现Comparable接口,会默认调用compareTo方法,故可往TreeSet中添加。
TreeSet <String>ts=new TreeSet<String>();
ts.add("a");
ts.add("c");
ts.add("b");
ts.add("2");
ts.add("1");
System.out.println(ts);
输出结果为[12abc]。
同样的还有诸如Integer、Data类,也默认都实现了Comparable接口。
2 往TreeSet集合中添加自定义对象
当往TreeSet中添加自定义对象实现排序(按自己的方式排序)时,有两种方法。
1>让元素自身具备比较功能:需要实现Comparable接口,覆盖compareTo()方法
2>让集合自身具备比较功能:需要额外定义一个类,实现Comparator接口,重写compare(t1,t2)方法,该类对象作为参数传递给TreeSet集合中的构造函数。
排序规则:先按汽车的名称排序,若相同,按轮子数排序,再相同就不能往里添加了。
第一种方式:
class Car implements Comparable{
String name;
int viechle;
public Car(String name,int viechle){
this.name=name;
this.viechle=viechle;
}
public String getNameViechle() {
return name+" "+viechle;
}
public void setNameViechle(String name,int viechle) {
this.name = name;
this.viechle=viechle;
}
public int compareTo(Object o){
//System.out.println("调用compareTo函数");
Car car=(Car)o;
int temp=this.name.compareTo(car.name);
return temp==0?this.viechle-car.viechle:temp;
}
}
---------------------------------------------------------
public class TreeSetDemo {
public static void main(String[] args) {
// TODO Auto-generated method stub
Set <Car>ts=new TreeSet<Car>();
ts.add(new Car("c",2));
ts.add(new Car("a",3));
ts.add(new Car("a",1));
ts.add(new Car("b",2));
ts.add(new Car("a",1));
Iterator<Car> it=ts.iterator();
while(it.hasNext()){
Car car=it.next();
System.out.println(car.getNameViechle());
}
}
}
输出结果:
a 1
a 3
b 2
c 2
第二种方式:
public class comparatorDemo implements Comparator<Car> {
@Override
public int compare(Car o1, Car o2) {
// TODO Auto-generated method stub
//System.out.println("调用compare方法");
Car c1=(Car)o1;
Car c2=(Car)o2;
int temp=c1.compareTo(c2);
return temp==0?c1.viechle-c2.viechle:temp;
}
}
----------
public class TreeSetDemo {
public static void main(String[] args) {
// TODO Auto-generated method stub
Set <Car>ts=new TreeSet<Car>(new comparatorDemo());//TreeSet(Comparator<? super E> comparator)
ts.add(new Car("c",2));
ts.add(new Car("a",3));
ts.add(new Car("a",1));
ts.add(new Car("b",2));
Iterator<Car> it=ts.iterator();
while(it.hasNext()){
Car car=it.next();
System.out.println(car.getNameViechle());
}
}
}
输出结果:
a 1
a 3
b 2
c 2