TreeSet是一个排序的、无重复元素的集合。
如果创建自定义对象,并且TreeSet创建时调用的是无参的构造函数,那么必须要实现comparable接口。
示例代码:向TreeSet中插入无参的自定义对象Persons(含name,age)。
注意:
1、首先要实现comparable接口
2、comparable接口内容要合理,比如只判断年龄进行排序,那么结果是年龄相同的所有Persons对象,将被视为是重复的,所以如果姓名也要按照字母顺序排序,那么也要在判断年龄相同时,给出判断name的情况,否则TreeSet将会视为重复元素,不予插入。
package test;
import java.util.Iterator;
import java.util.TreeSet;
public class TestTreeSet {
public static void main(String[] args) {
TreeSet ts = new TreeSet<>();
ts.add(new Persons("java01",1));
ts.add(new Persons("java03",3));
ts.add(new Persons("java02",2));
ts.add(new Persons("java04",4));
ts.add(new Persons("java05",4));
for(Iterator it = ts.iterator();it.hasNext();){
System.out.println(it.next());
}
}
}
class Persons implements Comparable{
String name;
int age;
Persons(String name,int age){
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 int compareTo(Object obj) {
if(!(obj instanceof Persons))
throw new RuntimeException();
Persons p = (Persons)obj;
if((this.age-p.age) == 0)
return this.name.compareTo(p.name);
else
return this.age-p.age;
}
@Override
public String toString() {
return "姓名:"+this.name+",年龄:"+this.age;
}
}