TreeSet

TreeSet:

不保证有序，可以对元素进行排序，二叉树结构，不同步的

 

排序方式：需要元素具备比较功能，所以元素需要实现Comparable接口，覆盖compareTo方法
如何保证元素唯一性?

 

其实参考的就是比较方法compareTo的返回值是否是0，是0，就是重复元素，不存。

 

 

 

需求中也有这样一种情况，元素具备的比较功能不是所需要的，也就是说不想按照
自然排序的方法，而是按照自定义的排序方式，对元素进行排序。
而且，存储到TreeSet中的元素万一没有比较功能，该如何排序呢？
这时，就只能使用第二种比较方式---是让集合具备比较功能，定义一个比较器

 

 

比较器更为灵活，自然排序通常作为元素的默认排序

 

创建TreeSet集合的两种方法：

1.存入TreeSet集合中的元素对象实现Comparable接口，重写compareTo方法，使传入的对象具备比较功能

2.自定义类实现Coparator接口，重写compare方法，在创建Set对象的时候，将实现了Comparator接口的comparator对象传入Set的构造方法中，使定义的TreeSet集合具备比较功能。

 

 

public class TreeSetDemo {

	public static void main(String[] args) {
		Set set = new TreeSet(new ComparatorByName());
		
		set.add(new Student("xiaoming",26));
		set.add(new Student("xiaoming",27));
		set.add(new Student("xiaoqiang",15));//java.lang.ClassCastException因为学生要排序，就需要比较，而没有定义比较方法，无法完成排序
		set.add(new Student("daniu",21));
		set.add(new Student("xiaoniu",21));
		
		for (Iterator it = set.iterator(); it.hasNext();) {
			Student stu = (Student)it.next();
			System.out.println(stu.getName()+"::"+stu.getAge());
		}
	}
}

 

class Student implements Comparable{
	private String name;
	private int age;
	public Student() {
		super();
	}
	public Student(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 "NewPeople [name=" + name + ", age=" + age + "]";
	}
	
	//覆盖hashCode方法
	@Override
	public int hashCode() {
		final int prime = 31;
		int result = 1;
		result = prime * result + age;
		return result;
	}
	
	//覆盖equals方法
	@Override
	public boolean equals(Object obj) {
		if (this == obj)
			return true;
		if (obj == null)
			return false;
		if (getClass() != obj.getClass())
			return false;
		Student other = (Student) obj;
			if(!name.equals(other.name)){
				return false;
			}
		
		return true;
	}
	/*
	 *学生就具备了比较功能，改功能是自然 排序使用的方法 
	 */
	@Override
	public int compareTo(Object o) {
		if(this==o)
			return 0;
		Student stu = (Student)o;
//		System.out.println(this.name+this.age+"----------------"+stu.name+stu.age);
//		if(this.age>stu.age)
//			return 1;
//		if(this.age<stu.age)
//			return -1;
		return this.age-stu.age==0? this.name.compareTo(stu.name) : this.age-stu.age;
	}
}

 

import java.util.Comparator;
/**
 * 自定义了一个比较器，用来对学生对象按照姓名进行排序
 * @author dell
 *
 */

//实现Comparator接口，覆盖compare方法，将Comparator接口的对象，
//作为参数传递给TreeSet集合的构造方法

public class ComparatorByName implements Comparator {

	@Override
	public int compare(Object o1, Object o2) {
		Student s1 = (Student)o1;
		Student s2 = (Student)o2;
		
		int temp = s1.getName().compareTo(s2.getName());
		
		return temp==0?s1.getAge()-s2.getAge():temp;
	}

}

关注我的微信公众号（曲健磊的个人随笔），观看更多精彩内容：