自定义类对象利用TreeSet排序

2019-01-29

利用TreeSet类和java.lang.Comparable接口实现给对象集合排序

TreeSet继承关系
treeset-map

TreeSet排序规则

使用二叉树的原理对新add()的对象按照指定的顺序排序（升序、降序），每增加一个对象都会进行排序，将对象插入的二叉树指定的位置。
Integer和String对象都可以进行默认的TreeSet排序，而自定义类的对象是不可以的，自己定义的类必须实现Comparable接口，并且覆写相应的compareTo()函数，才可以正常使用。

//实现接口
public class Model implements Comparable<Model>{
	private int age;
	private String name;
	
	public Model(int age,String name){
		this.age=age;
		this.name=name;
	}
	//重写方法
	@Override
	public int compareTo(Model o) {
		if(this.age>o.age){
			return 1;
		}else if(this.age<o.age){
			return -1;
		}else{
			return 0;
		}
	}
	
	//省略getter和setter方法...
}

main方法

public static void main(String[] args) {
	Set<Model> tree=new TreeSet<>();
	Model m1=new Model(1,"1");
	Model m3=new Model(3,"3");
	Model m2=new Model(2,"2");
	Model m5=new Model(5,"5");
	Model m4=new Model(4,"4");
	
	tree.add(m1);
	tree.add(m3);
	tree.add(m2);
	tree.add(m5);
	tree.add(m4);
	Iterator<Model> it=tree.iterator();
	while(it.hasNext()){
		System.out.print(it.next().getAge()+" ");
	}
}

输出结果为

1 2 3 4 5