Java 动态数组的实现示例

目录

• 静态数组
• 动态数组的实现原理
  • 1.添加元素
  • 2.删除元素
  • 3.数组扩容
  • 4.数组缩减

int[] array = new int[6];
for (int i = 0; i < array.length; i++){
    array[i] = 2 * i + 1;
}

public class Array<E> {
    private E[] data;       // 使用静态数组存放数组元素
    private int size;       // 记录数组元素数量
 
    public Array(int capacity) {
        this.data = (E[]) new Object[capacity];
        this.size = 0;
    }
 
    public Array() {
        this(10);   // 默认capacity为10
    }
 
    // 数组扩容/缩减
    public void resize(int newCapacity) {
        // 新数组长度必须大于0
        if (newCapacity < 0) throw new IllegalArgumentException("capacity must > 0!");
        // 创建新数组
        E[] newData = (E[]) new Object[newCapacity];
        // 将原数组元素放入新数组中
        for (int i = 0; i < size; i++) {
            newData[i] = data[i];
        }
        // 将引用指向新数组
        data = newData;
    }
 
    /**
     * 在指定位置添加元素
     * 指定位置处的元素需要向右侧移动一个单位
     * @param index   索引
     * @param element 要添加的元素
     */
    public void add(int index, E element) {
        if (index < 0 || index > size) throw new IllegalArgumentException("Illegal index, index must > 0 and <= size!");
        // 数组满员触发扩容
        if (size == data.length) {
            resize(2 * data.length);  // 扩容为原数组的2倍
        }
        // 从尾部开始，向右移动元素，直到index
        for (int i = size - 1; i >= index; i--) {
            data[i + 1] = data[i];
        }
        // 添加元素
        data[index] = element;
        size++;
    }
 
    // 数组头部添加元素
    public void addFirst(E element) {
        add(0, element);
    }
 
    // 数组尾部添加元素
    public void addLast(E element) {
        add(size, element);
    }
 
    /**
     * 删除指定位置元素
     * 通过向左移动一位，覆盖指定位置处的元素，实现删除元素(data[size - 1] = null)
     * @param index 索引
     */
    public E remove(int index) {
        if (index < 0 || index > size) throw new IllegalArgumentException("Illegal index, index must > 0 and < size!");
        // 数组长度为0时抛出异常
        if (size == 0) throw new IllegalArgumentException("Empty array!");
        E removedElement = data[index];
        // 向左移动元素
        for (int i = index; i < size - 1; i++) {
            data[i] = data[i + 1];
        }
        // 将尾部空闲出的位置置为空，释放资源
        data[size - 1] = null;
        size--;
        // size过小触发数组缩减
        if (size == data.length / 4 && data.length / 2 != 0) resize(data.length / 2);
        return removedElement;
    }
 
    // 删除头部元素
    public E removeFirst() {
        return remove(0);
    }
 
    // 删除尾部元素
    public E removeLast() {
        return remove(size - 1);
    }
 
    // 重写Override方法，自定义数组显示格式
    @Override
    public String toString() {
        StringBuilder str = new StringBuilder();
        // 显示数组的整体情况(长度、总容量)
        str.append(String.format("Array: size = %d, capacity = %d
[", size, data.length));
        // 循环添加数组元素至str
        for (int i = 0; i < size; i++) {
            str.append(data[i]);
            if (i < size - 1) str.append(", ");
        }
        str.append("]");
        return str.toString();
    }
}

public static void main(String[] args) {
        // 添加10个元素
        Array<Integer> arr = new Array<>();
        for (int i = 0; i < 10; i++)
            arr.add(i, i);
        // 查看数组当前状态
        System.out.println(arr);
        // 继续添加元素，观察是否扩容
        arr.add(arr.size, 7);
        System.out.println(arr);
 
        // 再删除6个元素，观察是否缩减
        for (int i = 0; i < 6; i++) {
            System.out.println("元素" + arr.removeFirst() + "已被删除！");
        }
        System.out.println(arr);
    }
 
/*
输出结果：
Array: size = 10, capacity = 10
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
Array: size = 11, capacity = 20
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 7]
元素0已被删除！
元素1已被删除！
元素2已被删除！
元素3已被删除！
元素4已被删除！
元素5已被删除！
Array: size = 5, capacity = 10
[6, 7, 8, 9, 7]
*/

    // 获取数组长度
    public int getSize() {
        return size;
    }
 
    // 获取数组总容量
    public int getCapacity() {
        return data.length;
    }
 
    // 判断数组是否为空
    public boolean isEmpty() {
        return getSize() == 0;
    }
 
    // 查找指定元素在数组中的位置
    public int search(E element) {
        for (int i = 0; i < getSize(); i++) {
            if (data[i].equals(element)) {
                return i;
            }
        }
        // -1表示未找到
        return -1;
    }
 
    // 判断指定元素是否在数组中
    public boolean contains(E element) {
        return search(element) != -1;
    }
 
    // 按照索引查找元素值
    public E get(int index) {
        if (index < 0 || index > size) throw new IllegalArgumentException("Illegal index, index must > 0 and < size!");
        return data[index];
    }
 
    // 查找头部元素
    public E getFirst() {
        return get(0);
    }
 
    // 查找尾部元素
    public E getLast() {
        return get(getSize() - 1);
    }
 
    // 设置指定位置的元素值
    public void set(int index, E element) {
        if (index < 0 || index > size) throw new IllegalArgumentException("Illegal index, index must > 0 and < size!");
        data[index] = element;
    }
 
    /**
     * 按照元素值删除
     * 只删除数组中第一个元素值与指定值相等的元素
     * @param element 指定元素值
     */
    public boolean removeElement(E element) {
        int index = search(element);
        if (index != -1) {
            remove(index);
            return true;
        }
        return false;
    }
 
    /**
     * 按照元素值删除
     * 删除数组中所有值与指定值相等的元素
     *
     * @param element 指定元素值
     */
    public boolean removeElementAll(E element) {
        boolean isRemoved = false;
        int i = getSize() - 1;
        while (i >= 0) {
            if (data[i].equals(element)) {
                remove(i);
                isRemoved = true;
            }
            i--;
        }
        return isRemoved;
    }