我用Java学数据结构之动态数组底层实现 ^-^

动态数组,见名知意,不像静态数组那样,一旦开辟了空间,就不能再进行增加或者删除操作,动态数组可以在数组存储满的时候自动扩容,数组中存储元素过少的时候自动缩容。

在这里,为了使用方便起见,使用了泛型

    /**
     * 动态数组中的元素存储于data中
     */
    private E[] data;
    /**
     * size记录数组中的元素个数
     */
    private int size;

首先实现add,传入要插入的索引位置,以及元素,在add的基础上可以实现addLast 以及addFirst ,以便后续在动态数组上实现栈,以及队列的操作

    public void add(int index, E e) {
        if (index < 0 || index > size) {
            throw new IllegalArgumentException("Add failed ,required index>=0 and index<=size ");
        }

        if (size == data.length) {
            resize(size * 2);
        }
        //将需要插入的位置(包括本身)向后挪动一位
        for (int i = size - 1; i >= index; i--) {
            data[i + 1] = data[i];
        }
        //最后将想要添加的元素放入想要放入的位置
        data[index] = e;
        size++;
    }

    public void addLast(E e) {
        add(size, e);
    }

    public void addFirst(E e) {
        add(0, e);
    }

 以及remove某个索引的元素,在这个基础上实现removeElement,removeFirst,removeLast

     /**
     * 从数组中删除指定索引位置元素并返回
     * ret 保存需要返回的元素       然后将 index+1至 size-1 统一向前挪动一个位置
     * 如果数组现存的元素个数不到数组长度的 1/4 就将数组缩容 为 原长的1/2
     *
     * @param index
     * @return
     */
    public E remove(int index) {
        if (index < 0 || index >= size) {
            throw new IllegalArgumentException("Remove failed , index illegal");
        }
        E ret = data[index];
        for (int i = index + 1; i < size; i++) {
            data[i - 1] = data[i];
        }
        //data[size - 1] = null;
        size--;
        //如果数组中存储的元素不及数组容量的1/4,缩小数组容量为现在的一半
        if (size == data.length / 4 && data.length / 2 != 0) {
            resize(data.length / 2);
        }
        return ret;
    }

    /**
     * 首先找到元素e的索引,再删除
     *
     * @param e
     */
    public void removeElement(E e) {
        int index = find(e);
        if (index != -1) {
            remove(index);
        }
    }

    public E removeLast() {
        return remove(size - 1);
    }

    public E removeFirst() {
        return remove(0);
    }

还有一个最重要的resize,为数组重新开辟空间

    /**
     * 将原数组的元素复制进新的数组中
     *
     * @param newCapacity
     */
    private void resize(int newCapacity) {
        E[] newData = (E[]) new Object[newCapacity];
        for (int i = 0; i < size; i++) {
            newData[i] = data[i];
        }
        data = newData;
    }

 

整个动态数组包装的方法:

 以及整个动态数组的代码:

public class Array<E> {

    /**
     * 动态数组中的元素存储于data中
     */
    private E[] data;
    /**
     * size记录数组中的元素个数
     */
    private int size;

    /**
     * 传入数组的容量capacity来构造Array
     *
     * @param capacity
     */
    public Array(int capacity) {
        data = (E[]) new Object[capacity];
        size = 0;
    }

    /**
     * 无参构造,默认数组容量capacity = 10
     */
    public Array() {
        this(10);
    }

    /**
     * 将传入的数组转化为动态数组存储
     *
     * @param arr
     */
    public Array(E[] arr) {
        data = (E[]) new Object[arr.length];
        for (int i = 0; i < arr.length; i++) {
            data[i] = arr[i];
        }
        size = arr.length;
    }

    public int getSize() {
        return size;
    }

    public int getCapacity() {
        return data.length;
    }

    public boolean isEmpty() {
        return size == 0;
    }

    public void addLast(E e) {
        add(size, e);
    }

    public void addFirst(E e) {
        add(0, e);
    }

    public void add(int index, E e) {
        if (index < 0 || index > size) {
            throw new IllegalArgumentException("Add failed ,required index>=0 and index<=size ");
        }

        if (size == data.length) {
            resize(size * 2);
        }
        //将需要插入的位置(包括本身)向后挪动一位
        for (int i = size - 1; i >= index; i--) {
            data[i + 1] = data[i];
        }
        //最后将想要添加的元素放入想要放入的位置
        data[index] = e;
        size++;
    }

    /**
     * 返回index位置的元素
     *
     * @param index
     * @return
     */
    public E get(int index) {
        if (index < 0 || index >= size) {
            throw new IllegalArgumentException("Get failed , index illegal");
        }
        return data[index];
    }

    public E getLast() {
        return get(size - 1);
    }

    public E getFirst() {
        return get(0);
    }

    /**
     * 修改index位置的元素e
     *
     * @param index
     * @param e
     */
    public void set(int index, E e) {
        if (index < 0 || index >= size) {
            throw new IllegalArgumentException("Set failed , index illegal");
        }
        data[index] = e;
    }

    /**
     * 查找数组中是否有元素e
     *
     * @param e
     * @return
     */
    public boolean contains(E e) {
        for (int i = 0; i < size; i++) {
            if (data[i].equals(e)) {
                return true;
            }
        }
        return false;
    }

    /**
     * 返回元素e的索引,如果没找到,返回-1
     *
     * @param e
     * @return
     */
    public int find(E e) {
        for (int i = 0; i < size; i++) {
            if (data[i].equals(e)) {
                return i;
            }
        }
        return -1;
    }

    public E removeLast() {
        return remove(size - 1);
    }

    public E removeFirst() {
        return remove(0);
    }

    /**
     * 首先找到元素e的索引,再删除
     *
     * @param e
     */
    public void removeElement(E e) {
        int index = find(e);
        if (index != -1) {
            remove(index);
        }
    }

    /**
     * 从数组中删除指定索引位置元素并返回
     * ret 保存需要返回的元素       然后将 index+1至 size-1 统一向前挪动一个位置
     * 如果数组现存的元素个数不到数组长度的 1/4 就将数组缩容 为 原长的1/2
     *
     * @param index
     * @return
     */
    public E remove(int index) {
        if (index < 0 || index >= size) {
            throw new IllegalArgumentException("Remove failed , index illegal");
        }
        E ret = data[index];
        for (int i = index + 1; i < size; i++) {
            data[i - 1] = data[i];
        }
        //data[size - 1] = null;
        size--;
        //缩小数组长度
        if (size == data.length / 4 && data.length / 2 != 0) {
            resize(data.length / 2);
        }
        return ret;
    }

    @Override
    public String toString() {
        StringBuilder result = new StringBuilder();
        result.append(String.format("DynamicArray : size = %d , capacity = %d\n", size, data.length));
        result.append('[');
        for (int i = 0; i < size; i++) {
            result.append(data[i]);
            if (i != size - 1) {
                result.append(", ");
            }
        }
        result.append(']');
        return result.toString();
    }

    /**
     * 交换索引为 i  和  j  的元素位置
     *
     * @param i
     * @param j
     */
    public void swap(int i, int j) {
        if (i < 0 || i >= size || j < 0 || j >= size) {
            throw new IllegalArgumentException("Index  is  illegal . ");
        }
        E t = data[i];
        data[i] = data[j];
        data[j] = t;
    }

    /**
     * 将原数组的元素复制进新的数组中
     *
     * @param newCapacity
     */
    private void resize(int newCapacity) {
        E[] newData = (E[]) new Object[newCapacity];
        for (int i = 0; i < size; i++) {
            newData[i] = data[i];
        }
        data = newData;
    }
}

最后测试下:

public class Main {

    public static void main(String[] args) {
        Array<Integer> array = new Array();
        for (int i = 0; i < 10; i++) {
            array.addLast(i);
        }
        System.out.println(array);

        array.add(1, 100);
        System.out.println(array);

        array.addFirst(-1);
        System.out.println(array);
        //[-1, 0, 100, 1, 2, 3, 4, 5, 6, 7, 8, 9]
        array.remove(2);
        System.out.println(array);

        array.removeElement(4);
        System.out.println(array);

        array.removeFirst();
        System.out.println(array);
    }
}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值