[19]-容器(集合)_Set_TreeSet的使用及底层源码

1）TreeMap（key，唯一，而且有序，升序），底层数据结构 黑红树，而且 TreeMap 中的 key 实际上就是一个 TreeSet，如 果使用自定义对象作为 key,要求必须具备比较规则

2）使用 TreeMap 要求使用内部比较器或外部比较器

使用api和list相同

代码中定义类内部比较器

public class Person implements Comparable {
  private String name;//姓名
  private int 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 Person(String name, int age) {
    super();
    this.name = name;
    this.age = age;
  }
  public Person() {
    super();
  }
  @Override
  public String toString() {
    return "Person [name=" + name + ", age=" + age + "]";
  }
  @Override
  public int compareTo(Object o) {
    //按照年龄的升序比较
    Person p=(Person)o;

    return this.age-p.getAge();
  }

}

定义外部比较器,第一版(扔存在相同元素只存一个)

外部比较器可以有多个

public class ComCharactor implements Comparator{

  @Override
  public int compare(Object o1, Object o2) {
    Person p1=(Person)o1;
    Person p2=(Person)o2;
    //compareTo是String类中指供的比较大小的方法
    return p1.getName().compareTo(p2.getName());
  }

}

外部比较器第二版

import java.util.Comparator;

public class ComCharactorAndAge implements Comparator {

  @Override
  public int compare(Object o1, Object o2) {
    Person p1=(Person)o1;//向下类型转换
    Person p2=(Person)o2;
    if(p1.getName().compareTo(p2.getName())<0){
      return -1;
    }else if(p1.getName().compareTo(p2.getName())>0){
      return 1;
    }else{
      return p1.getAge()-p2.getAge();
    }
    //return 0;
  }

}

测试代码’

import java.util.Comparator;
import java.util.NavigableMap;
import java.util.TreeMap;
import java.util.TreeSet;

/** 
 * TreeSet的底层所使用的是TreeMap   -->红黑树
 * TreeSet JDK源码分析
 *  private transient NavigableMap<E,Object> m;  //一个接口,间接继承了Map
 *    private static final Object PRESENT = new Object();
 *  public TreeSet() {
        this(new TreeMap<E,Object>());  //创建了一个TreeMap的对象
    }
    //调用的本类中的带参构造方法
    TreeSet(NavigableMap<E,Object> m) {
        this.m = m;   //接口new 实现类
    }
     public TreeSet(Comparator<? super E> comparator) {
        this(new TreeMap<>(comparator));  //调用了本类中的带参构造方法
    }
    添加方法
     public boolean add(E e) {
        return m.put(e, PRESENT)==null;  //传入的元素作为Map中的key，统一的值为Object类型的对象
    }
    public int size() {
        return m.size();
    }


 * @author Administrator
 *
 */
public class TestTreeSet {
  public static void main(String[] args) {

    //创建集合对象
    //TreeSet ts=new TreeSet();
    //Comparator comc=new ComCharactor();//创建外部比较器对象
    Comparator com=new ComCharactorAndAge();
    TreeSet ts=new TreeSet(com);
    //创建Person对象
    Person p1=new Person("marry", 20);
    Person p2=new Person("lili", 19);
    Person p3=new Person("jack", 20);
    Person p4=new Person("marry", 18);
    //添加到集合中
    ts.add(p1);
    ts.add(p2);
    ts.add(p3);
    ts.add(p4);
    System.out.println("集合中元素的个数:"+ts.size());
    System.out.println(ts);
  }
}