【java学习—十】TreeSet集合（5）

---

1. TreeSet

    TreeSet 是 SortedSet 接口的实现类， TreeSet 可以确保集合元素处于排序状态。
    TreeSet 支持两种排序方法：自然排序和定制排序。默认情况下， TreeSet 采用自然排序。

1.1. 自然排序

    默认情况下， TreeSet 采用自然排序。自然排序： TreeSet 会调用集合元素的 compareTo(Object obj) 方法来比较元素之间的大小关系，然后将集合元素按升序排列：
    • 如果 this > obj, 返回正数 1
    • 如果 this < obj, 返回负数 -1
    • 如果 this = obj, 返回 0 ，则认为这两个对象相等
    • 必须放入同样类的对象 ，( 默认会进行排序 ) 否则可能会发生类型转换异常 ，可以使用泛型来进行限制。

举例：

package day10;

import java.util.Set;
import java.util.TreeSet;

public class Test4 {
	public static void main(String[] args) {
		Set<Integer> set = new TreeSet<Integer>();
		//TreeSet自然排序
		set.add(5);
		set.add(2);
		set.add(4);
		set.add(3);
		
		System.out.println(set);
	}
}

运行结果：

举例2：实现其他功能

package day10;

import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;

public class Test4 {
	public static void main(String[] args) {
		Set<Integer> set = new TreeSet<Integer>();
		
		set.add(5);
		set.add(2);
		set.add(4);
		set.add(3);
		
		System.out.println(set);
		
		set.remove(5);
		set.contains(3);
//		set.clear();//清空集合
		
		//使用迭代器遍历集合
		Iterator<Integer> it = set.iterator();
		while(it.hasNext()) {
			System.out.println(it.next());
		}
		
		//for each迭代集合，推荐这种
		for(Integer i : set) {
			System.out.println(i);
		}
	}
}

1.2. 定制排序

    如果需要实现定制排序，则需要在创建 TreeSet 集合对象时，提供一个 Comparator 接口的实现类对象。由该 Comparator对象负责集合元素的排序逻辑。

举例：

package day10;

import java.util.Comparator;
import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;

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

		Person p1 = new Person("张三", 23);
		Person p2 = new Person("李四", 20);
		Person p3 = new Person("王五", 16);
		Person p4 = new Person("杨六", 29);
		
		Set<Person> set = new TreeSet<Person>(new Person());
		set.add(p1);
		set.add(p2);
		set.add(p3);
		set.add(p4);
		
		//for each迭代集合
		for(Person p : set) {
			System.out.println(p.name + " " + p.age);
		}
	
	}
}

class Person implements Comparator<Person>{//把person对象存到TreeSet中并按照年龄排序
	int age;
	String name;
	public Person() {
		
	}
	
	public Person(String name, int age) {
		this.name = name;
		this.age = age;
	}
	
	@Override
	public int compare(Person o1, Person o2) {//年龄正序排列
		if(o1.age > o2.age) {
			return 1;
		}else if(o1.age < o2.age) {
			return -1;
		}else {
			return 0;
		}	
	}
	
}

运行结果：