_101_Java_TreeSet的定制排序

------------------------------------------------

 

----------------------------------------------

 

 --------------------------------------------------

/**1、向TreeSet中添加的元素必须是同一个类的。
 * *2、可以添加进集合中的元素按照指定的顺序排列，像String,包装类等
 * 默认按照从小到大的顺序遍历。
 * 3、当向TreeSet中添加自定义类的对象时，有两种排序方法：①自然排序和②定制排序。
 * 4、自然排序，要求自定义类实现java.lang.Comparable接口并重写compareTo方法。
 * 5、向TreeSet中添加元素时，首先按照compareTo方法进行比较，一旦返回0，虽然仅是两个对象的属性值相同，但是程序会
 * 认为两个对象 是相同的，进而后一个对象不能添加进入TreeSet中。
 *
 *compareTo方法与equals方法、hashCode方法保持一致。
 * */ 

--------------------------------------------------

 PersonA.java

public class PersonA implements Comparable<Object>{

	private String name;
	private Integer age;
	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 PersonA() {
	}
	
	
	public PersonA(String name, int age) {
		super();
		this.name = name;
		this.age = age;
	}
	@Override
	public String toString() {
		return "PersonA [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;
		PersonA other = (PersonA) 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类的时候，依据此方法确定按照哪个属性排序。
*/	@Override
	public int compareTo(Object o) {
		if(o instanceof PersonA) {
			PersonA p = (PersonA) o;
			int i=this.getName().compareTo(p.getName());
			if(i==0) {
				return this.getAge().compareTo(p.getAge());
			}
			return i;
		}
		return 0;
	}
	
}

 

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

import org.junit.Test;

/**1、向TreeSet中添加的元素必须是同一个类的。
 * *2、可以添加进集合中的元素按照指定的顺序排列，像String,包装类等
 * 默认按照从小到大的顺序遍历。
 * 3、当向TreeSet中添加自定义类的对象时，有两种排序方法：①自然排序和②定制排序。
 * 4、自然排序，要求自定义类实现java.lang.Comparable接口并重写compareTo方法。
 * 5、向TreeSet中添加元素时，首先按照compareTo方法进行比较，一旦返回0，虽然仅是两个对象的属性值相同，但是程序会
 * 认为两个对象 是相同的，进而后一个对象不能添加进入TreeSet中。
 *
 *compareTo方法与equals方法、hashCode方法保持一致。
 * */
public class _002_TreeSet {

	public static void main(String[] args) {
		
		
	}
	
	@Test
	public void testA() {
		Set set = new TreeSet();
		set.add("AA");
		set.add("AA");
		set.add("JJ");
		set.add("GG");
		set.add("MM");
		Iterator it = set.iterator();
		while(it.hasNext()) {
			System.out.println(it.next());
		}
	}
	
	@Test
	public void testB() {
		Set set = new TreeSet();
		/*当PersonA类没有实现Comparable接口时向TreeSet中添加Person对象时，
		就会报ClassCastException*/
		set.add(new PersonA("DD",34));
		set.add(new PersonA("AA",35));
		set.add(new PersonA("CC",37));
		set.add(new PersonA("BB",39));
		set.add(new PersonA("EE",33));
		set.add(new PersonA("EE",33));
		set.add(new PersonA("EE",34));
		Iterator it = set.iterator();
		while(it.hasNext()) {
			System.out.println(it.next());
		}
		
	}
	
	
}

testA方法的输出结果：

 

 testB方法的输出结果：