Java底层基于二叉搜索树实现集合和映射/集合Set功能详解

/**
 * 集合的接口
 */
public interface Set<E> {
  void add(E e);//添加 <——<不能添加重复元素
  void remove(E e);//移除
  int getSize();//获取大小
  boolean isEmpty();//是否为空
  boolean contains(E e);//是否包含元素
  
}

//基于BST二分搜索树实现的集合Set
public class BSTSet<E extends Comparable<E>> implements Set<E> {//元素E必须满足可比较的
 
  //基于BST类的对象
  private BST<E> bst;
 
  //构造函数
  public BSTSet() {
    bst = new BST<>();
  }
 
  //返回集合大小
  @Override
  public int getSize() {
    return bst.size();
  }
 
  //返回集合是否为空
  @Override
  public boolean isEmpty() {
    return bst.isEmpty();
  }
 
  //Set添加元素
  @Override
  public void add(E e) {
    bst.add(e);
  }
 
  //是否包含元素
  @Override
  public boolean contains(E e) {
    return bst.contains(e);
  }
 
  //移除元素
  @Override
  public void remove(E e) {
    bst.remove(e);
  }
}

public static void main(String[] args) {
 
    System.out.println("Pride and Prejudice");
    //新建一个ArrayList存放单词
    ArrayList<String> words1=new ArrayList<>();
    //通过这个方法将书中所以单词存入word1中
    FileOperation.readFile("pride-and-prejudice.txt",words1);
    System.out.println("Total words : "+words1.size());
 
    BSTSet<String> set1=new BSTSet<>();
    //增强for循环，定一个字符串word去遍历words
    //底层的话会把ArrayList words1中的值一个一个的赋值给word
    for(String word:words1)
      set1.add(word);//不添加重复元素
    System.out.println("Total different words : "+set1.getSize());
 
    System.out.println("-------------------");
    System.out.println("Pride and Prejudice");
    //新建一个ArrayList存放单词
    ArrayList<String> words2=new ArrayList<>();
    //通过这个方法将书中所以单词存入word1中
    FileOperation.readFile("a-tale-of-two-cities.txt",words2);
    System.out.println("Total words : "+words2.size());
 
    BSTSet<String> set2=new BSTSet<>();
    //增强for循环，定一个字符串word去遍历words
    //底层的话会把ArrayList words1中的值一个一个的赋值给word
    for(String word:words2)
      set2.add(word);//不添加重复元素
    System.out.println("Total different words : "+set2.getSize());
  }