4.30Java TreeSet的使用和底层实现

4.30Java TreeSet的使用和底层实现

TreeSet的概念

底层用TreeMap实现,通过key来存储Set的元素

特点:

  • TreeSet内部需要对存储的元素进行排序,对应类需要实现Comparable接口

  • 根据compareTo()方法比较对象的大小,进行内部排序

  • 要排序需要实现comparable接口,实现comparable接口必须重写里面的compareTo方法

定义一个Employee类:

package collection.set;

/*
因为key可以是任意对象,treemap是递增的方式排序
如果key的值为类会如何排序?
新的接口---comparable
*/
/*定义一个新的类*/
public class Employee implements Comparable<Employee>{

   //定义它的属性
   int id;
   String name;
   double salary;
   /*以Employee为key,按照salary排序--->需要实现comparable接口*/

   /*构造器*/
   public Employee(int id, String name, double salary){
       super();
       this.id = id;
       this.name = name;
       this.salary = salary;
  }

   /*重写toString方法*/
   @Override
   public String toString(){
       return "id:" +  id + ",name:" + name + ",salary:" + salary;
  }

   /*实现接口当中未实现的方法*/
   @Override
   public int compareTo(Employee o){
       //TODO Auto-generated method stub
       //负数 : 小于, 0 : 等于, 正数 : 大于--->1 -1 0代替
       //比较salary
       if (salary > o.salary){
           //返回值
           return 1;
      }else if (salary < o.salary){
           return -1;
      }else {
           //如果工资相等利用id排序
           if (this.id > o.id){
               return 1;
          }else if (this.id < o.id){
               return -1;
          }else {
               return 0; //相等
          }
      }
  }
}

Test类:

import java.util.TreeSet;

/**
* 测试TreeSet的使用
* @author Lucifer
*/
public class TestTreeSet {
   public static void main(String[] args) {

       //建一个Set对象
       Set<Integer> set = new TreeSet<>();

       set.add(300);
       set.add(200);
       set.add(600);

       //利用增强for循环进行遍历
       for (Integer m : set){
           //遍历集合里面的所有元素然后打印出来
           System.out.println(m);
      }

       /*在创建一个Employee对象*/
       Set<Employee> set2 = new TreeSet<>();
       set2.add(new Employee(100,"Lucifer",20000));
       set2.add(new Employee(50,"Harden",15000));
       set2.add(new Employee(150,"James",10000));

       /*循环添加内容*/
       for (Employee m : set2){
           //打印出结果
           System.out.println(m);
      }
  }
}

 

 

上一篇:Java中TreeSet添加重复元素问题


下一篇:集合(一)