黑马程序员------TreeSet集合框架存储自定义元素之排序Comparable与Comparator

------- <a href="http://www.itheima.com"target="blank">android培训</a>、<a href="http://www.itheima.com"target="blank">java培训</a>、期待与您交流！ ----------

TreeSet集合框架存储自定义元素之排序Comparable与Comparator

我们知道，往TreeSet中存入的元素是要具有比较行的，TreeSet是要默认为元素排序的，但是我们如果存储的元素对象自身不具备比较性怎么办？有两种方式来解决这个问题，就要用到Comparable与Comparator这两个接口

 

方式1：

元素实现Comparable接口，强制具备比较性，并复写compareTo方法。根据返回值，确定元素在TreeSet数据结构中的位置。 其中保证元素唯一性的方式：就是参考compareTo方法的结果是否为0，如果return 0，视为两个对象重复，不存。 

代码示例如下：

 

package com.gzj.jihe;
import java.util.*;
/*需求：
往TreeSet集合中存储自定义对象学生。
按照学生的年龄进行排序。

另，排序时，当主要条件相同时，一定判断一下次要条件。

*/

class TreeSetDemo 
{
	public static void main(String[] args) 
	{
		TreeSet ts = new TreeSet();

		ts.add(new Student("a",22));
		ts.add(new Student("b",20));
		ts.add(new Student("c",19));
		ts.add(new Student("d",19));
		ts.add(new Student("b",20));

		Iterator it = ts.iterator();
		while(it.hasNext())
		{
			Student stu = (Student)it.next();
			System.out.println(stu.getName()+"..."+stu.getAge());
		}
	}
}


class Student implements Comparable//该接口强制让学生具备比较性。
{
	private String name;
	private int age;

	Student(String name,int age)
	{
		this.name = name;
		this.age = age;
	}
	public String getName()
	{
		return name;

	}
	public int getAge()
	{
		return age;
	}
	public int compareTo(Object obj)//复写这个方法，底层自动调用
	{
		if(!(obj instanceof Student))
			throw new RuntimeException("不是学生对象");
		Student s = (Student)obj;//强转
		System.out.println(this.name+"....compareto....."+s.name);
		if(this.age>s.age)
			return 1;
		if(this.age==s.age)
		{
			return this.name.compareTo(s.name);//String类本身也继承了Comparable接口
			                                   //复写compareTo方法
		}
		return -1;	
	}
	
}

 

方式2：

当元素自身不具备比较性时，或者具备的比较性不是所需求时，我们可以采取让容器自身具备比较性的方式。即定义一个实现了Comparator接口的比较器，并覆盖compare方法，将比较器对象作为参数传递给TreeSet集合的构造函数，根据比较器对元素进行排序。这种方法较为灵活。

 注意：当两种排序都存在时，以比较器为主。

 

package com.gzj.jihe;
import java.util.*;

class Student implements Comparable//该接口强制让学生具备比较性。
{
	private String name;
	private int age;

	Student(String name,int age)
	{
		this.name = name;
		this.age = age;
	}

	public int compareTo(Object obj)
	{	
		if(!(obj instanceof Student))
			throw new RuntimeException("不是学生对象");
		Student s = (Student)obj;
		if(this.age>s.age)
			return 1;
		if(this.age==s.age)
		{
			return this.name.compareTo(s.name);
		}
		return -1;
		
	}

	public String getName()
	{
		return name;

	}
	public int getAge()
	{
		return age;
	}
}

class TreeSetDemo2 
{
	public static void main(String[] args) 
	{
		TreeSet ts = new TreeSet(new MyCompare());//获取比较器接口的子类对象，按照里面定义的方式排序

		ts.add(new Student("lisi02",22));
		ts.add(new Student("lisi02",21));
		ts.add(new Student("lisi007",20));
		ts.add(new Student("lisi09",19));
		ts.add(new Student("lisi06",18));
		ts.add(new Student("lisi06",18));
		ts.add(new Student("lisi007",29));


		Iterator it = ts.iterator();
		while(it.hasNext())
		{
			Student stu = (Student)it.next();
			System.out.println(stu.getName()+"..."+stu.getAge());
		}
	}
}

class MyCompare implements Comparator//定义的子类比较器实现 Comparator接口
{
	public int compare(Object o1,Object o2)//并复写compare(Object o1,Object o2)方法
	{
		Student s1 = (Student)o1;
		Student s2 = (Student)o2;

		int num = s1.getName().compareTo(s2.getName());//先按照姓名排序
		if(num==0)//姓名相同情况下按照年龄排序
		{
			return new Integer(s1.getAge()).compareTo(new Integer(s2.getAge()));
		}
		return num;
	}
}<span style="font-family: Arial, Helvetica, sans-serif; background-color: rgb(255, 255, 255);"> </span>

总结：两种对元素排序方式最大的区别归结为一句话，方式一是使要排序的元素自身具备比较性，而方式二是另存储元素的集合具备比较性。