Java集合框架之_TreeSet特性

 1.向TreeSet中添加是元素必须是同一类型的。否则会报错。
 2.可以按照添加进集合中的元素的指定的顺序来遍历；如：String，包装类等默认按照从小到大的顺序遍历
 3.当向TreeSet中添加自定义类的对象时，有两种排序方法：①自然排序 ②指定排序
  3.1自然排序  要求自定义类实现java.lang.Comparable接口并重写compareTo(Object obj)方法，在此方法中，指明按照自定义类的哪个属性进行排序

  3.2指定排序  首先：创建一个实现了Comparator接口的类对象

                        其次：将此对象做为形参传给TreeSet的构造器中

                        再次：将Comparator接口中compare()方法里面设计到的类的对象添加到TreeSet中

 4.向TreeSet中添加自定义对象时，首先按照CompareTo()方法比较，一旦返回0，虽然是两个对象的此属性值相同，但是程序会认为这两个对象是相同的，进而后一个对象就会添加不进来。

 注意：在使用TreeSet自然排序时，compareTo()与hashCode()以及equals()三个方法必须保持一致！

 在使用TreeSet指定排序时，Comparator接口里面的compare()与hashCode()以及equals()三个方法必须保持一致！

@Test
	public void testTreeSet() {
		/**
		 * 3.1自然排序
		 * 要求：被添加到TreeSet中的自定义类实现java.lang.Comparable接口并重写compareTo(Object obj)方法，在此方法中，指明
		 * 按照自定义类的哪个属性进行排序
		 */
		Set set = new TreeSet();
		//Person类必须实现Comparable接口并重写compareTo(Object obj)方法
		set.add(new Person("MM", 21));
		set.add(new Person("GG", 23));
		set.add(new Person("AA", 18));
		set.add(new Person("QQ", 12));
		set.add(new Person("MM", 12));
		
		Iterator it = set.iterator();
		while (it.hasNext()) {
			System.out.println(it.next());
		}
	}

 因为是Person 类的对象要添加到TreeSet中，所以Person类要实现 Comparable接口并重写 compareTo(Object obj)方法，看以下代码！

public class Person implements Comparable {
	
	private String name;
	private Integer age;
	//此处省略get、set方法和构造器···
	@Override
	public String toString() {
		return "Person [name=" + name + ", age=" + age + "]";
	}
	@Override
	public int hashCode() {
		final int prime = 31;
		int result = 1;
		result = prime * result + age;
		result = prime * result + ((name == null) ? 0 : name.hashCode());
		return result;
	}
	@Override
	public boolean equals(Object obj) {
		if (this == obj)
			return true;
		if (obj == null)
			return false;
		if (getClass() != obj.getClass())
			return false;
		Person other = (Person) obj;
		if (age != other.age)
			return false;
		if (name == null) {
			if (other.name != null)
				return false;
		} else if (!name.equals(other.name))
			return false;
		return true;
	}

	//当向TreeSet中添加Person类的对象时，依据此方法，确定按照那个属性排序
	@Override
	public int compareTo(Object o) {
		//先判断穿过来的Object类型的参数是否是Person类型的
		if (o instanceof Person) {
			Person p = (Person)o;
			//return -this.age.compareTo(p.age);//前面加负号是按照倒序排列
			int i = this.age.compareTo(p.age);
			if (i == 0) {//如果年龄一样就按照name排序
				return this.name.compareTo(p.name);
			} else {
				return i;
			}
		}
		return 0;
	}
}

 咱们接着看 指定排序怎么实现！

@Test
	public void testTreeSet2() {
		/**
		 * 3.2指定排序
		 * 1.创建一个实现了Comparator接口的类对象
		 * 2.将此对象做为形参传给TreeSet的构造器中
		 * 3.将Comparator接口中compare()方法里面设计到的类的对象添加到TreeSet中
		 */
		//第一步：创建一个实现了Comparator接口的类对象
		Comparator com = new Comparator() {
			@Override
			public int compare(Object o1, Object o2) {
				if (o1 instanceof Customer && o2 instanceof Customer) {
					Customer c1 = (Customer)o1;
					Customer c2 = (Customer)o2;
					int i = c1.getId().compareTo(c2.getId());
					if (i == 0) {//如果id一样就按照name排序
						return c1.getName().compareTo(c2.getName());
					}
					return i;
				}
				return 0;
			}
		};
		//第二步：将此对象做为形参传给TreeSet的构造器中
		Set set = new TreeSet(com);
		//第三步：将Comparator接口中compare()方法里面设计到的类的对象添加到TreeSet中
		set.add(new Customer("GG", 23));
		set.add(new Customer("MM", 21));
		set.add(new Customer("DD", 18));
		set.add(new Customer("JJ", 26));
		set.add(new Customer("WW", 30));
		set.add(new Customer("SS", 30));
		
		for (Object obj : set) {
			System.out.println(obj);
		}
	}

指定排序还有另外一种写法，就是把第一步和第二步合并在了一起，请看以下示例代码！

	@Test
	public void testTreeSet4() {
		//第一步：创建一个实现了Comparator接口的类对象
		//第二步：将此对象做为形参传递给TreeSet构造器中
		Set set = new TreeSet(new Comparator() {

			@Override
			public int compare(Object o1, Object o2) {
				if (o1 instanceof Customer && o2 instanceof Customer) {
					Customer c1 = (Customer)o1;
					Customer c2 = (Customer)o2;
					int i = c1.getId().compareTo(c2.getId());
					if (i == 0) {
						return c1.getName().compareTo(c2.getName());
					}
					return i;
				}
				return 0;
			}
			
		});
		//向TreeSet中添加在Comparator接口中Compare()方法里面设计的类对象
		set.add(new Customer("GG", 23));
		set.add(new Customer("MM", 21));
		set.add(new Customer("DD", 18));
		set.add(new Customer("JJ", 26));
		set.add(new Customer("WW", 30));
		set.add(new Customer("SS", 30));
		
		for (Object obj : set) {
			System.out.println(obj);
		}
	}

是不是很简单！不过说到Set集合，用的最多的就是HashSet，以上作为扩充知识，理解即可！