Java集合(三)TreeSet的使用

package com.pason.treeset;

import java.util.Comparator;
import java.util.TreeSet;


/*
 * TreeSet容器和HashSet一样都不能放重复的对象
 * 最主要的是TreeSet中的对象要求必须能自己排序(排序方式要自己定义)
 * String类已经实现Comparable，所以不用自己定义
 * 所以TreeSet使用的时候更注重Comparable的实现
 * 在比较的时候,如果比较结果返回0那么只能加入1个对象
 */
public class TreeSetDemo {
	public static void main(String[] args) {
		//如果容器中的元素不定义比较规则
		//这时候会报Exception in thread "main" 
		//java.lang.ClassCastException，原因是User类不会自己排序
		
		//我们要实现Comparable接口才能add
		//Java中有两种排序的比较器
		//1)java.lang.Comparable
		TreeSet<User> user = new TreeSet<User>();
		User user1 = new User("Pason",123);
		User user2 = new User("Wade",456);
		User user3 = new User("Lee",789);
		user.add(user1);
		user.add(user2);
		user.add(user3);
		user.add(user2);
		System.out.println(user);
		//2)java.util.Comparator
		//TreeSet构造的时候,把Comparator作为构造器的参数
		
	/**
	 *  第二种写法(匿名内部类，也可以在外部写一个类实现Comparator接口,然后在本类内new一个对象传入TreeSet构造器) 
	 *  TreeSet<User> user= new TreeSet<User>(new Comparator<User>() {
     *
	 *			public int compare(User arg0, User arg1) {
	 *				return arg0.getName().compareTo(arg1.getName());
	 *			}
	 *		} );
	 *	User user1 = new User("Pason",123);
	 *	User user2 = new User("Wade",456);
	 *	User user3 = new User("Lee",789);
	 *	user.add(user1);
	 *	user.add(user2);
	 *	user.add(user3);
	 *	System.out.println(user);
     *
   */
	  
	 
	}

}

package com.pason.treeset;

class User implements Comparable<User> {
	private String name;
	private int id;
	
	public User(String name, int id) {
		super();
		this.name = name;
		this.id = id;
	}
	public String getName() {
		return name;
	}
	public void setName(String name) {
		this.name = name;
	}
	public int getId() {
		return id;
	}
	public void setId(int id) {
		this.id = id;
	}
	@Override
	public String toString() {
		return "Student [name=" + name + ", id=" + id + "]";
	}
	//重写compareTo方法
	public int compareTo(User arg0) {
		//自己定义，只要不同就要返回非0值
		if (!this.name.equals(arg0.name))
			 return this.name.compareTo(arg0.name);
		return 0;
	}
	
	

}