java 集合框架-TreeSet

/*
TreeSet可以对Set集合中的对象进行排序；

原理，TreeSet在底层调用了Comparable接口的方法compareTo().
所以，在使用TreeSet存储自定义对象时，要想实现自定义类对象的自然排序，需要类实现Comparable接口，并重写覆盖接口唯一的方法compareTo().

需求：
往TreeSet集合中存储自定义对象学生。
想按照学生的年龄进行排序。

记住：排序时，当主要条件相同时，一定判断一下次要条件。
在本例中，出现了两个同学的年龄是一样的，如果不进行判断次要条件动作，那么会有的学生的信息存储不上，应为在compareTo时返回的是0.
所以，需要继续判断姓名是否一致。在String类中也实现了Comparable接口中的compareTo方法。所以可以直接使用str.compareTo(str1)来实现字符串的比较。

TreeSet的底层数据结构是二叉树的。保存数据唯一性的依据是compareTo方法返回0.

TreeSet排序的第一种方式：让元素自身具备比较性。
元素需要实现Comparable接口，覆盖compareTo方法。
这种方式也称为元素的自然顺序，或者叫做默认顺序。

TreeSet排序的第二种方式：元素本身不具备比较性，那就构造比较器，比较器作为参数传递给TreeSet含参的构造函数。
当两种方式都存在时，以比较器为主。
！！！！！！！！！！！！两种方式的实现在程序TreeSetDemo1.java中！！！！！！！！！！！！！！！！！！
*/

import java.util.*;

class Student implements Comparable//因为TreeSet会对存的对象进行自然排序，其实是它实现了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;
		TreeSetDemo.sop(this.name+"......"+s.name);
		TreeSetDemo.sop(this.name);   //首先是Lily_01进来并存储在TreeSet集合中。然后是Lily_02进来并与Lily_01进行比较。
		if(this.age>s.age)            //底层调用的是compareTo方法。所以在lily_02进来后，this值得就是Lily_02。说白了就是谁进来this就代表谁。
			return 1;         //大的放在二叉树的右边
		if(this.age==s.age)
		{
			return this.name.compareTo(s.name);  //如果说是一样的话，就不会存储相同的对象。
		}
		return -1;  //小的放在二叉树的左边。
	}
}

class TreeSetDemo
{
	public static void main(String[] args)
	{
		TreeSet ts=new TreeSet();
		ts.add(new Student("Lily_01",10));
		ts.add(new Student("Lily_02",9));
		//ts.add(new Student("Lily_03",15));
		//ts.add(new Student("Lily_04",18));
		//ts.add(new Student("Lily_05",5));
		//ts.add(new Student("Lily_06",18));
		//ts.add(new Student("Lily_07",20));

		Iterator it=ts.iterator();
		while(it.hasNext())
		{
			Object obj=it.next();
			Student s=(Student)obj;

			//sop(s.getName()+".."+s.getAge());
		}
		System.out.println("Hello World!");
	}

	public static void sop(Object obj)
	{
		System.out.println(obj);
	}
}

/*
总结，TreeSet可以对存入的对象进行自然顺序的排序，
是因为其底层数据结构是二叉树，并且实现了Comparable接口的compareTo()方法。
通过自定义对象的类实现Comparable接口，并覆盖compareTo()方法，即可根据自己的需要存储
有顺序的对象数据。
返回1，说明新进元素较大，放在二叉树的右子节点；
返回-1，说明新进元素较小，放在二叉树的左子节点；
返回0，说明对象元素一致，则再比较其他次要条件（比如姓名一样，就比较年龄）

*/
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 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;
		int x=this.name.compareTo(s.name);
		if(x==0)
		{
			if(this.age>s.age)
				return 1;
			if(this.age==s.age)
				return 0;
			return -1;
		}
		return x;

		/*
		按照输入的顺序输出:
		return 1;
		*/

		/*
		按照输入的顺序逆序输出：
		return -1;
		*/

		/*
		只存储一个学生的信息：
		return 0;
		*/
	}

}

class TreeSetDemo1
{
	public static void main(String[] args)
	{

		TreeSet ts=new TreeSet();//使用方式一：通过实现Comparable接口并覆盖其中的compareTo()方法.
		TreeSet ts=new TreeSet(new MyCompare()); //使用方式二：使用比较器对年龄进行排序。构造比较器实现Comparator接口，并覆盖compare方法。

		ts.add(new Student("Lily",20));
		ts.add(new Student("Jhon",30));
		ts.add(new Student("Jane",19));
		ts.add(new Student("Sara",17));
		ts.add(new Student("Demetria",25));
		ts.add(new Student("Mike",25));
		ts.add(new Student("Lily",25));

		Iterator it=ts.iterator();
		while(it.hasNext())
		{
			Student s=(Student)it.next();
			sop(s.getName()+","+s.getAge());
		}

		System.out.println("Hello World!");
	}

	public static void sop(Object obj)
	{
		System.out.println(obj);
	}
}

class MyCompare implements Comparator  //定义比较器，实现按照学生年龄进行排序
{
	public int compare(Object obj1,Object obj2)
	{
		if(!(obj1 instanceof Student)||!(obj2 instanceof Student))
			throw new RuntimeException("有对象不是Student类型的。");
		Student s1=(Student)obj1;
		Student s2=(Student)obj2;
		if(s1.getAge()>s2.getAge())
			return 1;
		if(s1.getAge()==s2.getAge())
			return s1.getName().compareTo(s2.getName());
		return -1;
	}
}