Java集合框架-Set(TreeSet)

TreeSet： 不但可以去除重复元素，还可以对set集合中的元素进行排序， 底层数据结构是二叉树。简答归纳TreeSet保证元素的唯一性的依据：判断compareTo()方法是否return 0，如果返回0就认为是相同元素，就不会存储该元素。

下面创建一个TreeSet集合，集合中存放的是Intege类型的对象。

package com.test.list;
import java.util.Set;
import java.util.TreeSet;
public class TreeSetDemo
{
	public static void main(String[] args)
	{
		Set<Integer> set = new TreeSet<Integer>();
		set.add(3);
		set.add(2);
		set.add(3);
		set.add(5);
		set.add(9);
		System.out.println("Size = "+ set.size());
		System.out.println(set);		
	}
}

打印结果为：

Size = 4
[2, 3, 5, 9]

通过结果可以看出，集合中已将重复的元素去除，并且元素是按数字的大小顺序排列的。

下面在集合中存放一些自定义类型的元素：

package com.test.list;
import java.util.Set;
import java.util.TreeSet;
public class TreeSetDemo
{
	public static void main(String[] args)
	{
		Set<Person> set = new TreeSet<Person>();
		set.add(new Person("Linda", 33));
		set.add(new Person("kevin", 27));
		set.add(new Person("Joace", 25));
		System.out.println(set);		
	}
}
class Person
{
	private String name;
	private int 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;
	}
	public Person(String name, int age)
	{
		this.name = name;
		this.age = age;
	}
}

运行后会抛出异常

Exception in thread "main" java.lang.ClassCastException: com.test.list.Person cannot be cast to java.lang.Comparable
	at java.util.TreeMap.put(TreeMap.java:542)
	at java.util.TreeSet.add(TreeSet.java:238)
	at com.test.list.TreeSetDemo.main(TreeSetDemo.java:10)

之所以会出现该异常，是因为在创建类Person是没有指定比较算法，也就是集合中的元素不知道按照什么来比较大小，无法确定元素的存放位置。

使用TreeSet时必须指定比较的算法，指定的方式有两种：
自然顺序：要存储的类必须实现Comparable接口，并且复写compareTo()方法，在方法中指定算法。
比较器顺序：在创建TreeSet时，传入一个比较器Comparator，复写compare()方法，在该方法中指定算法。

//TreeSet排序的第一种方式：
//让元素自身具备比较性，元素需要实现Comparable接口，复写compareTo()方法，这种方式也称为自然顺序
package com.test.list;
import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;
public class TreeSetDemo
{
	public static void main(String[] args)
	{
		Set<Person> set = new TreeSet<Person>();
		set.add(new Person("Linda", 33));
		set.add(new Person("kevin", 27));
		set.add(new Person("Joace", 25));
		set.add(new Person("Joace", 36));
		for(Iterator<Person> it = set.iterator(); it.hasNext();)
		{
			Person person = it.next();
			System.out.println(person.getName()+", "+person.getAge());
		}
	}
}
//要让类Person实现Comparable接口
class Person implements Comparable<Person>
{
	private String name;
	private int 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;
	}
	public Person(String name, int age)
	{
		this.name = name;
		this.age = age;
	}
	//要复写compareTo()方法
	public int compareTo(Person o)
	{
		//按年龄的大小来排序
		int num = new Integer(this.age).compareTo(o.age);
		//当主要条件相同时，必须要比较次要条件，否则，当年龄相等时，则认为是重复元素被去除
		if(0 == num)
			return new Integer((this.name).compareTo(o.name));
		return num;
	}
}

/*
排序的第二种方式：
当元素自身不具备比较性，或者具备的比较性不是所需要的，这是可以让集合自身具备比较性，需要自己定义一个比较器。
比较器：定义一个类，实现Comparator接口，复写compare方法。
当两种比较方式都存在时，以比较器为主。
 * */
package com.test.list;
import java.util.Comparator;
import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;
public class TreeSetDemo
{
	public static void main(String[] args)
	{
		Set<Person> set = new TreeSet<Person>(new MyCompare());
		set.add(new Person("Linda", 33));
		set.add(new Person("kevin", 27));
		set.add(new Person("Joace", 33));
		set.add(new Person("Joace", 36));
		for(Iterator<Person> it = set.iterator(); it.hasNext();)
		{
			Person person = it.next();
			System.out.println(person.getName()+", "+person.getAge());
		}
	}
}
//要让类Person实现Comparable接口
class Person implements Comparable<Person>
{
	private String name;
	private int 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;
	}
	public Person(String name, int age)
	{
		this.name = name;
		this.age = age;
	}
	//要复写compareTo()方法
	public int compareTo(Person o)
	{
		//按年龄的大小来排序
		int num = new Integer(this.age).compareTo(o.age);
		//当主要条件相同时，必须要比较次要条件，否则，当年龄相等时，则认为是重复元素被去除
		if(0 == num)
			return new Integer((this.name).compareTo(o.name));
		return num;
	}
}
class MyCompare implements Comparator<Person>
{	//复写compare()方法
	public int compare(Person o1, Person o2)
	{
		//为了和自然顺序有区别，这里使用倒序
		int num = new Integer(o2.getAge()).compareTo(new Integer(o1.getAge()));
		return num != 0 ? num : o2.getName().compareTo(o1.getName());		
	}	
}