TreeSet(一)--排序

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对象

 public class People implements Comparable {  

     private String name;
     private String sex;
     private int age;  

     /*提供set和get方法，全参和无参构造方法*/  

     @Override
     public int compareTo(Object o) {  

         People people = (People) o;
         if (people.age > this.age) {  

             return 1;
         }
         if (this.age == people.age) {  

             return this.name.compareTo(people.name);
         }
         if ( people.age<this.age ) {  

             return -1;
         }  

         return 0;
     }
 }  

测试类

 public class SortTest {  

     public static void main(String[] args) {  

         People people1 = new People("小a", "男", 18);
         People people2 = new People("小a", "女", 16);
         People people3 = new People("小c", "女", 18);
         People people4 = new People("小b", "女", 22);
         People people5 = new People("小d", "男", 19);  

         Set<People> treeSet = new TreeSet<People>();  

         treeSet.add(people1);
         treeSet.add(people2);
         treeSet.add(people3);
         treeSet.add(people4);
         treeSet.add(people5);  

         Iterator iterator = treeSet.iterator();  

         while (iterator.hasNext()) {  

             People people = (People) iterator.next();  

             System.out.println("姓名：" + people.getName() + "\t年龄:" + people.getAge());
         }  

     }
 }  

运行结束：后台报错：

Exception in thread "main" java.lang.ClassCastException: com.treeset.sort.People cannot be cast to java.lang.Comparable

（3）举例一个学生有语文、数学、 英语三门课，按总分从高到底排序

     Student对象

 public class Student {  

     private String name;
     private int   chinese;
     private int   math;
     private int   english;  

     /*提供set和get方法，同时提供无参数，有参数构造方法*/  

     //同时单独要加上getSum方法
     public int getSum(){
         return this.chinese + this.english + this.math;
     }
 }  

测试类

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

 public class TreeTest2 {  

     public static void main(String[] args) {  

         Student student1=new Student("小明", 80, 90, 70);
         Student student2=new Student("小王", 60, 80, 90);
         Student student3=new Student("小钱", 100, 100, 80);
         Student student4=new Student("小徐", 20, 10, 90);
         Student student5=new Student("小李", 80, 80, 80);
         Student student6=new Student("小李", 70, 80, 90);  

         TreeSet<Student>  treeSet=new TreeSet(new MyComparable());  

         treeSet.add(student1);
         treeSet.add(student2);
         treeSet.add(student3);
         treeSet.add(student4);
         treeSet.add(student5);
         treeSet.add(student6);  

         Iterator<Student>  iterator=treeSet.iterator();  

         while(iterator.hasNext()){  

             Student student=iterator.next();  

             System.out.println(student.toString());
         }
     }
 }  

MyComparable类

 import java.util.Comparator;  

 public class MyComparable implements Comparator<Student>{  

     @Override
     public int compare(Student s1, Student s2) {  

          // 总分从高到低（注意这里是s2减s1）
         int num = s2.getSum() - s1.getSum();  

         if(num>0){
             return 1;
         }
         if(num<0){
             return -1;
         }  

         if(num==0){
             //这步非常关键，没有这个如果总成绩相同名字不同 ，那set集合就默认是相同元素，就会被覆盖掉
             return s2.getName().compareTo(s1.getName());
         }
             return 0;
     }
 }  

   /**
  * 是不是很奇怪为什么只有五条数据，而不是六条，那是因为有一条数据被覆盖了。
  * 你的Comparator实现类，是先比较成绩，成绩相同，在比较名字，那如果总成绩
  * 相同，姓名也相同，那不是默认是重复数据，TreeSet当然给你覆盖掉了。所以这
  * 里有个小李被覆盖掉了。那如何写才规范，下面这样就不会出现覆盖。
         */
 @Override
    // 创建一个TreeSet集合
    public int compare(Student s1, Student s2) {
        // 总分从高到低（注意这里是s2减s1）
        int num = s2.getSum() - s1.getSum();
        // 总分相同的不一定语文相同
        int num2 = num == 0 ? s1.getChinese() - s2.getChinese() : num;
        // 总分相同的不一定数学相同
        int num3 = num2 == 0 ? s1.getMath() - s2.getMath() : num2;
        // 总分相同的不一定英语相同
        int num4 = num3 == 0 ? s1.getEnglish() - s2.getEnglish() : num3;
        // 姓名还不一定相同
        int num5 = num4 == 0 ? s1.getName().compareTo(s2.getName()) : num4;
        return num5;
    }  

       最后在思考，set集合在开发中到底有啥用，好像我们开发当中要用集合一般会用ArrayList，好像很少用到TreeSet集合

      这个时候你就要想TreeSet集合的特点了：排序和唯一

 

 举个小例子：

     （4） 编写一个程序，获取10个1至20的随机数，要求随机数不能重复。

 import java.util.Iterator;
 import java.util.TreeSet;
 import java.util.Random;
 /*
  * 编写一个程序，获取10个1至20的随机数，要求随机数不能重复。
  *
  * 分析：
  *         A:创建随机数对象
  *         B:创建一个TreeSet集合
  *         C:判断集合的长度是不是小于10
  *             是：就创建一个随机数添加
  *             否：不搭理它
  *         D:遍历TreeSet集合
  */
 public class HashSetDemo {
     public static void main(String[] args) {
         // 创建随机数对象
         Random r = new Random();
         // 创建一个Set集合
         TreeSet<Integer> treeSet = new TreeSet<Integer>();
         // 判断集合的长度是不是小于10
         while (treeSet.size() < 10) {
             int x = r.nextInt(20) + 1;
             treeSet.add(x);
         }
         // 遍历Set集合
         for (int x : treeSet) {
             System.out.println(x);
         }
     }
 }  

     用这个例子，希望你能有扩散你的思维，也应该知道在什么时候用TreeSet集合了。

 

     

     水滴石穿，成功的速度一定要超过父母老去的速度！ 少尉【1】