黑马程序员------TreeSet的自然排序与比较器

------- android培训、java培训、期待与您交流！ ----------

使用TreeSet存储元素，要求元素必须具有"比较的功能"

  我们怎么才能具有比较的功能：

  方式一：将我们的类实现：Comparable接口，重写compareTo()-->自然排序；

  方式二：我们的类不用实现任何接口，在实例化TreeSet时，给TreeSet传递一个"比较器对象"，这个比较器对象一定要是：实现Comparator接口，重写compare()方法

使用自然排序:

创建一个自定义类在类中实现comparable接口  重写compareTo方法,这样在向TreeSet中存储元素的时候 会自动调用类的compareto方法进行比较

public class Student implements Comparable {
	private String name;
	private int age;
	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 "Student [name=" + name + ", age=" + age + "]";
	}
	@Override
	public int compareTo(Object o) {
		System.out.println("compareTo......");
		Student stu = (Student)o;
		
//这里是按照字典顺序比较名字字符串如果名字相同在比较年龄 然后返回比较结果
		int n = this.name.compareTo(stu.getName());
		int n2 = (n == 0 ? this.age - stu.age : n);//如果name相同，再比较年龄
		return n2;

使用比较器:

在创建集合的时候为该集合指定一个比较器:

TreeSet<Student> stuSet = new TreeSet<>(new MyComparator());

以下为比较器类:

public class MyComparator<T> implements Comparator<T> {
	@Override
	public int compare(T o1, T o2) {
		System.out.println("MyComparator-->compare()......");
		//将两个参数转换为Student
		Student stu1 = (Student)o1;
		Student stu2 = (Student)o2;
		
		//先按照字典顺序比较姓名在比较年龄
		int n = stu1.getName().compareTo(stu2.getName());
		int n2 = (n == 0 ? stu1.getAge() - stu2.getAge() : n);
		
		return n2;
	}

}

为了简化代码,还可以直接在创建集合的时候创建一个匿名内部类的比较器:

TreeSet<Student> ts=new TreeSet<>(new Comparator<Student>() {
			@Override
			public int compare(Student o1, Student o2) {
<span style="white-space:pre">			</span>int n = o1.getName().compareTo(o2.getName());
 <span style="white-space:pre">			</span>int n2=(n==0)?(o1.getAge()-o2.getAge()):

<span style="white-space:pre">			</span>n;return n2;}});