Java TreeSet集合、TreeSet集合自排序的方式、自定义的引用数据类型的排序实现、自然排序Comparable的使用、比较器排序Comparator的使用

TreeSet集合概述和特点

TreeSet集合特点

• 元素有序: 这里的顺序不是指存储和取出的顺序，而是按照一定的规则进行排序，具体排序方式取决于构造方法；

TreeSet():根据其元素的自然排序进行排序
TreeSet(Comparator comparator) :根据指定的比较器进行排序

• 没有带索引的方法， 所以不能使用普通for循环遍历，可使用迭代器和增强for循环；
• 由于是Set集合，所以不包含重复元素的集合；

TreeSet集合自排序的方式:

1. 有值特性的元素直接可以升序排序。(浮点型， 整型)
2. 字符串类型的元素会按照首字符的ASCII排序。
3. 对于自定义的引用数据类型，Treeset默认无法排序，执行的时候直接报错，因为人家不知道排序规则。

---

存储整数并遍历：

public class TreeSetTest {
	public static void main(String[] args) {
		TreeSet<Integer> ts = new TreeSet<>();

		ts.add(10);
		ts.add(40);
		ts.add(20);
		ts.add(98);
		ts.add(75);

		ts.add(40);

		for (Integer i : ts) {
			System.out.println(i);
		}

		/*
		 * 结果： 10 20 40 75 98
		 */
	}
}

---

自定义的引用数据类型的排序实现:

对于自定义的引用数据类型，TreeSet默认无法排序
所以我们需要定制排序的大小规则，程序员定义大小规则的方案有2种:

1. 直接为对象的类实现比较器规则接口Comparable，重写compareTo比较方法
   Java规则：
   如果程序员认为比较者大于被比较者，返回正数
   如果程序员认为比较者小于被比较者，返回负数
   如果程序员认为比较者等于被比较者，返回0
2. 直接为集合设置Comparator对象，重写比较方法
   java规则还是一样

实际开发在类上重写，或为集合设置比较器。
如果两种比较规则都存在，则会优先使用选择集合设置的比较器。

---

1. 自然排序Comparable的使用

存储学生对象并遍历，创建TreeSet集合使用无参构造方法

• 要求: 按照年龄从小到大排序，年龄相同时，按照姓名的字母顺序排序

public class Student implements Comparable<Student> {//实现接口
	private String name;
	private int age;

	public Student() {
	}

	public Student(String name, int age) {
		super();
		this.name = name;
		this.age = 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;
	}

	// 重写compareTo
	@Override
	public int compareTo(Student s) {
//		return 0;	//返回0则认为是同一个元素，不添加
//		return 1;	//返回正数是升序
//		return -1; //返回负数是倒序
		// 按照年龄从小到达排序
		return (this.age - s.age) == 0 ;
	}

}

public class TreeSetTest {
	public static void main(String[] args) {
		TreeSet<Student> ts = new TreeSet<>();
		Student s1 = new Student("a小明", 10);
		Student s2 = new Student("b小红", 21);
		Student s3 = new Student("c小李", 22);
		Student s4 = new Student("d小蓝", 10);

		Student s5 = new Student("d小蓝", 10);
		ts.add(s1);
		ts.add(s2);
		ts.add(s3);
		ts.add(s4);

		for (Student s : ts) {
			System.out.println(s.getName() + "," + s.getAge());
		}
		/*
		 * a小明,10 d小蓝,10 b小红,21 c小李,22
		 */
	}
}

结论

• 用TreeSet集合存储自定义对象， 无参构造方法使用的是自然排序对元素进行排序的；
• 自然排序， 就是让元素所属的类实现Comparable接口，重写compareTo(T o)方法；
• 重写方法时，一定要注意排序规则必须按照要求的主要条件和次要条件来写；

2.比较器排序Comparator的使用

• 存储学生对象并遍历，创建TreeSet集合使用带参构造方法
• 要求: 按照年龄从小到大排序，年龄相同时，按照姓名的字母顺序排序

public class Student {
	private String name;
	private int age;

	public Student() {
	}

	public Student(String name, int age) {
		super();
		this.name = name;
		this.age = 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 class TreeSetTest {
	public static void main(String[] args) {
		TreeSet<Student> ts = new TreeSet<>(new Comparator<Student>() {
			@Override
			public int compare(Student s1, Student s2) {
				// this.age - s.age
				// s1,s2
				return s1.getAge() - s2.getAge();
			}
		});

		Student s1 = new Student("a小明", 10);
		Student s2 = new Student("b小红", 21);
		Student s3 = new Student("c小李", 22);
		Student s4 = new Student("d小蓝", 10);

		Student s5 = new Student("d小蓝", 10);
		ts.add(s1);
		ts.add(s2);
		ts.add(s3);
		ts.add(s4);

		for (Student s : ts) {
			System.out.println(s.getName() + "," + s.getAge());
		}

	}
}

结论

• 用TreeSet集合存储自定义对象， 带参构造方法使用的是比较器排序对元素进行排序的；
• 比较器排序,就是让集合构造方法接收Comparator的实现类对象，重写compare(T o1,T o2)方法；
• 重写方法时， 一定要注意排序规则必须按照要求的主要条件和次要条件来写；