java: TreeSet

TreeSet

TreeSet可以确保元素处于排序状态
————————————————————支持两种排序：自然排序，定制排序，默认情况下是自然排序

自然排序：

必须放入同样类型的对象（默认会进行排序），否则会发生类型转换异常——————可以使用泛型来进行限制

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

public class Test {

	public static void main(String[] args) {

		Set<Integer> set = new TreeSet<Integer>();//<Integer>使用泛型限制了
		// TreeSet自然排序
		set.add(2);
		set.add(5);
		set.add(4);
		set.add(3);
		set.add(0);
		System.out.println(set);// [0, 2, 3, 4, 5]
		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);
		}

	}

}

定制排序

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

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

public class Test {


class Person implements Comparator<Person>{//把person对象存到Treeset中并且按照年龄进行排序
	
	int age;
	String name;
	
	public Person(){
		
	}
	public Person(int age,String name){
		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;
		}

		public static void main(String[] args) {
		
		Person p1=new Person(23,"张5");
		Person p2=new Person(21,"张1");
		Person p3=new Person(28,"张2");
		Person p4=new Person(29,"张3");
		Person p5=new Person(22,"张4");
		Set<Person> set=new TreeSet<Person>(new Person());
		set.add(p1);
		set.add(p2);
		set.add(p3);
		set.add(p4);
		set.add(p5);
		for (Person p:set){
			System.out.println(p.name+"  "+p.age);
		}
	}
}

张1  21
张4  22
张5  23
张2  28
张3  29