我们之前学习的类都无法对元素自动排序,java设置这个类能够将储存在集合中的元素自动排序,TreeSet类的底层使用的是二叉树结构。在储存对象时会根据对象的比较功能,确定对象的大小,从而确定该对象在二叉树中的位置。
import java.util.Iterator;
import java.util.TreeSet;
public class TreeSetDemo {
public static void main(String[] args) {
TreeSet ts=new TreeSet();
ts.add("aaaa");
ts.add("bbbb");
ts.add("cccc");
ts.add("dddd");
for (Iterator iterator = ts.descendingIterator(); iterator.hasNext();) {
System.out.println(iterator.next());
}
}
}
底层TreeSet类在对对像进行排序时其实是将对象强转成cmparable,然后通过comparableTo方法进行比较大小,从而确定位置。
java中规定要想某个类具有比较功能,就要实现comparable接口。所以当TreeSet类存放自定义类时,该类要实现comparable接口。
用TreeSet类存放自定义对象:
import java.util.Iterator;
import java.util.TreeSet;
class People implements Comparable {
private String name;
private int age;
public People(String name, int age) {
super();
this.name = name;
this.age = age;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public int getAge() {
return age;
}
public void setAge(int age) {
this.age = age;
}
@Override
public String toString() {
return "People [name=" + name + ", age=" + age + "]";
}
@Override
public int compareTo(Object o) {
// TODO Auto-generated method stub
People p = (People) o;
return this.age - p.age;
}
}
public class TreeSetDemo {
public static void main(String[] args) {
TreeSet ts = new TreeSet();
ts.add(new People("aaaa", 128));
ts.add(new People("m", 122));
ts.add(new People("aaaa", 122));
ts.add(new People("aana", 11));
for (Iterator iterator = ts.descendingIterator(); iterator.hasNext();) {
System.out.println(iterator.next());
}
}
}