数据结构学习之集合篇(Java)

集合(Set):
对应数学中集合的概念。集合中元素的三个特性：确定性，互异性，无序性。但是在计算机数据结构中， 也有有序集合（一般基于搜索树实现），无序集合（一般基于哈希表实现），多重集合（允许元素重复）的分类。我们在使用中一般都是利用其不允许存储重复元素的功能。
集合接口Set:增删查功能

public interface Set<E> {
     //添加元素
     void add(E e);
     //删除元素
     void remove(E e);
     //查询集合是否包含元素e
     boolean contains(E e);
     //查询集合中元素的个数
     int getSize();
     //集合是否为空
     boolean isEmpty();
}

基于链表实现的LinkedListSet:

public class LinkedListSet<E> implements Set<E> {

    private LinkedList<E> list;

    public LinkedListSet(){
        list = new LinkedList<>();
    }
    @Override
    public int getSize(){
        return list.getSize();
    }
    @Override
    public boolean isEmpty(){
        return list.isEmpty();
    }
    @Override
    public boolean contains(E e){
        return list.contains(e);
    }
    @Override
    public void add(E e){
        if (!list.contains(e)){
            list.addFirst(e);
        }
    }
    @Override
    public void remove(E e){
        list.removeElement(e);
    }
}

链表LinkedList:

public class LinkedList<E> {
    private class Node{
        public E e;
        public Node next;

        public Node(E e, Node next) {
            this.e = e;
            this.next = next;
        }
        public Node(E e) {
            this.e = e;
            this.next = null;
        }

        public Node() {
            this.e = null;
            this.next = null;
        }

        @Override
        public String toString() {
            return e.toString();
        }
    }
    //虚拟头结点
    private Node dummyHead;
    //链表中元素的个数
    private int size;

    public LinkedList(){
        dummyHead = new Node(null,null);
        size = 0;
    }
    //获取链表中元素的个数
    public int getSize(){
        return size;
    }
    //链表中元素个数是否为空
    public boolean isEmpty(){
        return size == 0;
    }
    //O(1)
    public void addFirst(E e){
        add(0,e);
    }
    //在链表的index(0-based)位置添加新的元素e
    //在链表中不是一个常用的操作,练习用
    public void add(int index,E e){
        if (index < 0 || index > size){
            throw new IllegalArgumentException("Add failed,Illegal index");
        }
            Node prev = dummyHead;
            //拿到要插入节点的前一个节点
            for (int i =0;i < index; i++){
                prev = prev.next;
            }
            Node node = new Node(e);
            node.next = prev.next;
            prev.next = node;
//          prev.next = new Node(e,prev.next)
            size ++;
    }
    //在链表末尾添加元素e
    public void addLast(E e){
        add(size,e);
    }
    // 获得链表的第index(0-based)个位置的元素
    //在链表中不是一个常用的操作,练习用
    public E get(int index){
        if (index < 0 || index >= size){
            throw new IllegalArgumentException("Add failed,Illegal index");
        }
        Node cur = dummyHead.next;
        for (int i =0;i < index; i++){
            cur = cur.next;
        }
        return cur.e;
    }
    //获得链表的第一个元素
    public E getFirst(){
        return get(0);
    }
    //获得链表的最后一个元素
    public E getLast(){
        return get(size - 1);
    }
    //修改第index位置的元素为e
    public void set(int index,E e){
        if (index < 0 || index >= size){
            throw new IllegalArgumentException("Add failed,Illegal index");
        }
        Node cur = dummyHead.next;
        for (int i =0;i < index; i++){
            cur = cur.next;
        }
        cur.e = e;
    }
    //查询链表中是否包含元素e
    public boolean contains(E e){
        //思路就是遍历链表中的元素然后与e进行比较
        Node cur = dummyHead.next;
        while (cur != null){
            if (cur.e.equals(e)){
                return true;
            }else {
                cur = cur.next;
            }
        }
        return false;
    }
    //删除在index位置的元素
    //在链表中不是常用的操作，练习用
    public E remove(int index){
        //先判断参数是否合法
        if (index < 0 || index >= size){
            throw new IllegalArgumentException("Add failed,Illegal index");
        }
        Node prev = dummyHead;
        for (int i = 0; i < index; i ++){
            prev = prev.next;
        }
        Node delNode = prev.next;
        prev.next = delNode.next;
        delNode.next = null;
        size --;

        return delNode.e;
    }
    //删除链表的头结点
    public E removeFirst(){
       return remove(0);
    }
    //删除链表的尾节点
    public E removeLast(){
        return remove(size - 1);
    }

    //从链表中删除元素e
    public void removeElement(E e){
        Node prev = dummyHead;
        while (prev.next != null){
            if (prev.next.e.equals(e))
                break;;
            prev = prev.next;
        }
        if (prev.next != null){
            Node delNode = prev.next;
            prev.next = delNode.next;
            delNode.next = null;
        }
    }
    //重写toString方法
    @Override
    public String toString(){
        StringBuilder res = new StringBuilder();
        //遍历打印
        res.append("dummyHead ->");
        for (Node cur = dummyHead.next; cur != null; cur = cur.next){
            res.append(cur+"->");
        }
        res.append("NUll");
        return res.toString();
    }
}

基于上一章实现的二分搜索树实现BSTSet:

public class BSTSet<E extends Comparable<E>> implements Set<E> {

    private BSTNormal<E> bst;
    //覆盖构造方法
    public BSTSet(){
        bst = new BSTNormal<>();
    }

    @Override
    public int getSize(){
        return bst.getSize();
    }

    @Override
    public boolean isEmpty(){
        return bst.isEmpty();
    }
    /*因为我们实现的二分搜索树是默认不包含重复元素的（重复元素被覆盖掉了），要包含重复元素实现起来反而要麻烦一些，所以我们可以
      直接使用bst的add方法来实现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);
    }
}