TreeSet方法之一 当向TreeSet中添加Person对象 续 当有多个属性值时的自然排序

package andycpp;

public class Person implements Comparable {
	private String name;
	private Integer age;
	//get set 方法
	public String getName() {
		return name;
	}
	public void setName(String name) {
		this.name = name;
	}
	public Integer getAge() {
		return age;
	}
	public void setAge(Integer age) {
		this.age = age;
	}
	//无参构造器
	public Person() {
		super();
	}
	//带参构造器
	public Person(String name, Integer age) {
		super();
		this.name = name;
		this.age = age;
	}
	//toString
	@Override
	public String toString() {
		return "Person [name=" + name + ", age=" + age + "]";
	}
	@Override
	public int hashCode() {
		final int prime = 31;
		int result = 1;
		result = prime * result + ((age == null) ? 0 : age.hashCode());
		result = prime * result + ((name == null) ? 0 : name.hashCode());
		return result;
	}
	@Override
	public boolean equals(Object obj) {
		if (this == obj)
			return true;
		if (obj == null)
			return false;
		if (getClass() != obj.getClass())
			return false;
		Person other = (Person) obj;
		if (age == null) {
			if (other.age != null)
				return false;
		} else if (!age.equals(other.age))
			return false;
		if (name == null) {
			if (other.name != null)
				return false;
		} else if (!name.equals(other.name))
			return false;
		return true;
	}
	
	//重写了未实现的抽象方法，方法实现了什么功能：就是你希望按照哪个属性来排列
	//当向TreeSet中添加Person类的对象时，依据此方法，确定按照哪个属性排列
	//Person有两个属性name,age,这里按照age 排序
	
	@Override
	public int compareTo(Object o) {
		if(o instanceof Person){ //保证是Person类型的
			Person p = (Person)o;
//			return this.name.compareTo(p.name);//name已经重写了compareTo
			return this.age.compareTo(p.age);//从小到大
//			return -this.age.compareTo(p.age);//从大到小
						
			
		} 
		return 0; //如果不是这个类型，就返回0，
	}
	
}

package andycpp;

import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;

import org.junit.Test;

/**
 * Collection接口：
 * 		List接口
 * 				ArrayList(主要的实现类)
 * 				LinkedList(对于频繁的插入删除操作)
 * 				Vector(古老的实现类，线程安全，但效率要低于ArrayList)
 * 		Set接口	存储无序，不可重复的元素.Set中常用的方法都是Collection下定义的。
 * 				HashSet(主要实现类)
 * 				LinkedHashSet
 * 				TreeSet
 * */
public class TestSet {
	/*
	 * TreeSet:
	 * 1,向TreeSet中添加的元素必须是同一个类的。类型相同
	 * 2,可以按照添加进集合中的元素的指定的顺序遍历。像String,包装类等默认按照从小到大的顺序遍历。
	 * 3,当向TreeSet中添加自定义类的对象时，有两种排序方法： 自然排序和定制排序
	 * 4，自然排序：要求自定义类实现java.lang.Comparable接口并重写其compareTo(Object obj)方法
	 * 在此方法中，指明按照自定义类的哪个属性进行排序。
	 * 
	 * 5，向TreeSet中添加元素时，首先按照compareTo()进行比较，一旦返回0，虽然仅是两个对象的此属性值相同，
	 * 但是程序会认为这两个对象是相同的，进而后一个对象就不能添加进来。
	 * compareTo()与hashCode()以及equals()三者保持一致。
	 */
	@Test
	public void testTreeSet1(){
		Set set = new TreeSet();
		/*
		 *  当Person类没有实现Comparable接口时，当向TreeSet中添加Person对象时，报ClassCastException错误
		 *  
		 * */
		set.add(new Person("CC",23));
		set.add(new Person("MM",21));
		set.add(new Person("GG",25));
		set.add(new Person("JJ",24));
		set.add(new Person("KK",20));
		set.add(new Person("DD",20));
	
		

		for(Object str :set){
			System.out.println(str);
		}


		
		
	}
}

输出

Person [name=KK, age=20]
Person [name=MM, age=21]
Person [name=CC, age=23]
Person [name=JJ, age=24]
Person [name=GG, age=25]

发现并没有输出DD,20

再看修改了Person类之后的程序

package andycpp;

public class Person implements Comparable {
	private String name;
	private Integer age;
	//get set 方法
	public String getName() {
		return name;
	}
	public void setName(String name) {
		this.name = name;
	}
	public Integer getAge() {
		return age;
	}
	public void setAge(Integer age) {
		this.age = age;
	}
	//无参构造器
	public Person() {
		super();
	}
	//带参构造器
	public Person(String name, Integer age) {
		super();
		this.name = name;
		this.age = age;
	}
	//toString
	@Override
	public String toString() {
		return "Person [name=" + name + ", age=" + age + "]";
	}
	@Override
	public int hashCode() {
		final int prime = 31;
		int result = 1;
		result = prime * result + ((age == null) ? 0 : age.hashCode());
		result = prime * result + ((name == null) ? 0 : name.hashCode());
		return result;
	}
	@Override
	public boolean equals(Object obj) {
		if (this == obj)
			return true;
		if (obj == null)
			return false;
		if (getClass() != obj.getClass())
			return false;
		Person other = (Person) obj;
		if (age == null) {
			if (other.age != null)
				return false;
		} else if (!age.equals(other.age))
			return false;
		if (name == null) {
			if (other.name != null)
				return false;
		} else if (!name.equals(other.name))
			return false;
		return true;
	}
	
	//重写了未实现的抽象方法，方法实现了什么功能：就是你希望按照哪个属性来排列
	//当向TreeSet中添加Person类的对象时，依据此方法，确定按照哪个属性排列
	//Person有两个属性name,age,这里按照age 排序
	
	@Override
	public int compareTo(Object o) {
		if(o instanceof Person){ //保证是Person类型的
			Person p = (Person)o;
//			return this.name.compareTo(p.name);//name已经重写了compareTo
//			return this.age.compareTo(p.age);//从小到大
//			return -this.age.compareTo(p.age);//从大到小
			
			//先定义一个变量来获取这个值
			int i = this.age.compareTo(p.age);
			if(i==0){//说明这个属性值两者一样，再比较剩下的属性
				return this.name.compareTo(p.name);
			}else{
				return i;
			}
		} 
		return 0; //如果不是这个类型，就返回0，
	}
	
}

再执行TestSet.java程序

输出

Person [name=DD, age=20]
Person [name=KK, age=20]
Person [name=MM, age=21]
Person [name=CC, age=23]
Person [name=JJ, age=24]
Person [name=GG, age=25]