TreeSet(一)
一、TreeSet定义：
与HashSet是基于HashMap实现一样，TreeSet同样是基于TreeMap实现的。
1）TreeSet类概述
使用元素的自然顺序对元素进行排序
或者根据创建 set 时提供的 Comparator 进行排序
具体取决于使用的构造方法。
2）TreeSet是如何保证元素的排序和唯一性的
底层数据结构是红黑树(红黑树是一种自平衡的二叉树)
（1） 自然排序
import java.util.TreeSet; /** TreeSet集合的特点：排序和唯一 * * 通过观察TreeSet的add()方法，我们知道最终要看TreeMap的put()方法。 */ public class TreeTest1 { public static void main(String[] args) { // 创建集合对象 // 自然顺序进行排序 TreeSet<Integer> treeSet = new TreeSet<Integer>(); // 创建元素并添加 treeSet.add(8); treeSet.add(4); treeSet.add(5); treeSet.add(6); treeSet.add(6); // 遍历 for(Integer i : treeSet){ System.out.print(i); } } } /*后台输出：4568 重复会被覆盖*/
（2）如果存入对象怎么排序，记住如果是对象一定要重写Comparator方法
People对象
1 public class People implements Comparable { 2 3 private String name; 4 private String sex; 5 private int age; 6 7 /*提供set和get方法，全参和无参构造方法*/ 8 9 @Override 10 public int compareTo(Object o) { 11 12 People people = (People) o; 13 if (people.age > this.age) { 14 15 return 1; 16 } 17 if (this.age == people.age) { 18 19 return this.name.compareTo(people.name); 20 } 21 if ( people.age<this.age ) { 22 23 return -1; 24 } 25 26 return 0; 27 } 28 }
测试类
1 public class SortTest { 2 3 public static void main(String[] args) { 4 5 People people1 = new People("小a", "男", 18); 6 People people2 = new People("小a", "女", 16); 7 People people3 = new People("小c", "女", 18); 8 People people4 = new People("小b", "女", 22); 9 People people5 = new People("小d", "男", 19); 10 11 Set<People> treeSet = new TreeSet<People>(); 12 13 treeSet.add(people1); 14 treeSet.add(people2); 15 treeSet.add(people3); 16 treeSet.add(people4); 17 treeSet.add(people5); 18 19 Iterator iterator = treeSet.iterator(); 20 21 while (iterator.hasNext()) { 22 23 People people = (People) iterator.next(); 24 25 System.out.println("姓名：" + people.getName() + " 年龄:" + people.getAge()); 26 } 27 28 } 29 }
运行结束：后台报错：
Exception in thread "main" java.lang.ClassCastException: com.treeset.sort.People cannot be cast to java.lang.Comparable
（3）举例一个学生有语文、数学、 英语三门课，按总分从高到底排序
Student对象
1 public class Student { 2 3 private String name; 4 private int chinese; 5 private int math; 6 private int english; 7 8 /*提供set和get方法，同时提供无参数，有参数构造方法*/ 9 10 11 //同时单独要加上getSum方法 12 public int getSum(){ 13 return this.chinese + this.english + this.math; 14 } 15 }
测试类
1 import java.util.Iterator; 2 import java.util.TreeSet; 3 4 public class TreeTest2 { 5 6 public static void main(String[] args) { 7 8 Student student1=new Student("小明", 80, 90, 70); 9 Student student2=new Student("小王", 60, 80, 90); 10 Student student3=new Student("小钱", 100, 100, 80); 11 Student student4=new Student("小徐", 20, 10, 90); 12 Student student5=new Student("小李", 80, 80, 80); 13 Student student6=new Student("小李", 70, 80, 90); 14 15 16 TreeSet<Student> treeSet=new TreeSet(new MyComparable()); 17 18 treeSet.add(student1); 19 treeSet.add(student2); 20 treeSet.add(student3); 21 treeSet.add(student4); 22 treeSet.add(student5); 23 treeSet.add(student6); 24 25 Iterator<Student> iterator=treeSet.iterator(); 26 27 while(iterator.hasNext()){ 28 29 Student student=iterator.next(); 30 31 System.out.println(student.toString()); 32 } 33 } 34 }
MyComparable类
1 import java.util.Comparator; 2 3 public class MyComparable implements Comparator<Student>{ 4 5 @Override 6 public int compare(Student s1, Student s2) { 7 8 // 总分从高到低（注意这里是s2减s1） 9 int num = s2.getSum() - s1.getSum(); 10 11 if(num>0){ 12 return 1; 13 } 14 if(num<0){ 15 return -1; 16 } 17 18 if(num==0){ 19 //这步非常关键，没有这个如果总成绩相同名字不同 ，那set集合就默认是相同元素，就会被覆盖掉 20 return s2.getName().compareTo(s1.getName()); 21 } 22 return 0; 23 } 24 }
1 /** 2 * 是不是很奇怪为什么只有五条数据，而不是六条，那是因为有一条数据被覆盖了。 3 * 你的Comparator实现类，是先比较成绩，成绩相同，在比较名字，那如果总成绩 4 * 相同，姓名也相同，那不是默认是重复数据，TreeSet当然给你覆盖掉了。所以这 5 * 里有个小李被覆盖掉了。那如何写才规范，下面这样就不会出现覆盖。 6 */ 7 @Override 8 // 创建一个TreeSet集合 9 public int compare(Student s1, Student s2) { 10 // 总分从高到低（注意这里是s2减s1） 11 int num = s2.getSum() - s1.getSum(); 12 // 总分相同的不一定语文相同 13 int num2 = num == 0 ? s1.getChinese() - s2.getChinese() : num; 14 // 总分相同的不一定数学相同 15 int num3 = num2 == 0 ? s1.getMath() - s2.getMath() : num2; 16 // 总分相同的不一定英语相同 17 int num4 = num3 == 0 ? s1.getEnglish() - s2.getEnglish() : num3; 18 // 姓名还不一定相同 19 int num5 = num4 == 0 ? s1.getName().compareTo(s2.getName()) : num4; 20 return num5; 21 }
最后在思考，set集合在开发中到底有啥用，好像我们开发当中要用集合一般会用ArrayList，好像很少用到TreeSet集合
这个时候你就要想TreeSet集合的特点了：排序和唯一
举个小例子：
（4） 编写一个程序，获取10个1至20的随机数，要求随机数不能重复。
1 import java.util.Iterator; 2 import java.util.TreeSet; 3 import java.util.Random; 4 /* 5 * 编写一个程序，获取10个1至20的随机数，要求随机数不能重复。 6 * 7 * 分析： 8 * A:创建随机数对象 9 * B:创建一个TreeSet集合 10 * C:判断集合的长度是不是小于10 11 * 是：就创建一个随机数添加 12 * 否：不搭理它 13 * D:遍历TreeSet集合 14 */ 15 public class HashSetDemo { 16 public static void main(String[] args) { 17 // 创建随机数对象 18 Random r = new Random(); 19 // 创建一个Set集合 20 TreeSet<Integer> treeSet = new TreeSet<Integer>(); 21 // 判断集合的长度是不是小于10 22 while (treeSet.size() < 10) { 23 int x = r.nextInt(20) + 1; 24 treeSet.add(x); 25 } 26 // 遍历Set集合 27 for (int x : treeSet) { 28 System.out.println(x); 29 } 30 } 31 }
用这个例子，希望你能有扩散你的思维，也应该知道在什么时候用TreeSet集合了。
水滴石穿，成功的速度一定要超过父母老去的速度！ 少尉【1】