手撕代码 - Java用数组实现ArrayList

作者：逍遥Sean
简介：一个主修Java的Web网站\游戏服务器后端开发者
主页：https://blog.csdn.net/Ureliable
觉得博主文章不错的话，可以三连支持一下~ 如有疑问和建议，请私信或评论留言！

前言
当我们在Java中使用ArrayList时，其背后的核心实现是基于动态数组的。本文将介绍如何使用纯数组来实现一个简化版的ArrayList，主要涵盖了元素添加、获取、删除以及动态扩容等基本功能。

1. 数据结构和初始设定

我们首先定义一个类 MyArrayList，它包含以下几个关键属性：

• data：用于存储元素的数组。
• size：当前存储的元素个数。

public class MyArrayList<E> {
    private static final int INITIAL_CAPACITY = 10;
    private Object[] data;
    private int size;
    
    public MyArrayList() {
        this.data = new Object[INITIAL_CAPACITY];
        this.size = 0;
    }
    
    // 省略其他代码
}

在构造函数中，我们初始化了一个容量为10的Object数组，并将初始大小设为0。

2. 添加元素的方法

我们需要实现一个方法 add 来向数组中添加元素。如果数组空间不足，我们需要自动扩展数组的容量。

public void add(E element) {
    ensureCapacity(size + 1);
    data[size++] = element;
}

private void ensureCapacity(int minCapacity) {
    if (minCapacity > data.length) {
        int newCapacity = data.length * 2;
        if (newCapacity < minCapacity) {
            newCapacity = minCapacity;
        }
        data = Arrays.copyOf(data, newCapacity);
    }
}

在 add 方法中，我们首先调用 ensureCapacity 方法来确保数组容量足够。如果数组容量不足，我们将数组扩展为原来的两倍大小，并将新元素添加到数组末尾。

3. 获取元素的方法

为了从ArrayList中获取元素，我们需要实现 get 方法，根据索引返回对应位置的元素。

public E get(int index) {
    if (index < 0 || index >= size) {
        throw new IndexOutOfBoundsException("Index: " + index + ", Size: " + size);
    }
    return (E) data[index];
}

4. 删除元素的方法

我们还需要实现 remove 方法来删除指定位置的元素，并将后续元素向前移动一个位置。

public void remove(int index) {
    if (index < 0 || index >= size) {
        throw new IndexOutOfBoundsException("Index: " + index + ", Size: " + size);
    }
    int numMoved = size - index - 1;
    if (numMoved > 0) {
        System.arraycopy(data, index + 1, data, index, numMoved);
    }
    data[--size] = null;
}

5. 完整实现

import java.util.Arrays;

public class MyArrayList<E> {
    private static final int INITIAL_CAPACITY = 10;
    private Object[] data;
    private int size;
    
    public MyArrayList() {
        this.data = new Object[INITIAL_CAPACITY];
        this.size = 0;
    }
    
    public void add(E element) {
        ensureCapacity(size + 1);
        data[size++] = element;
    }
    
    public E get(int index) {
        if (index < 0 || index >= size) {
            throw new IndexOutOfBoundsException("Index: " + index + ", Size: " + size);
        }
        return (E) data[index];
    }
    
    public void remove(int index) {
        if (index < 0 || index >= size) {
            throw new IndexOutOfBoundsException("Index: " + index + ", Size: " + size);
        }
        int numMoved = size - index - 1;
        if (numMoved > 0) {
            System.arraycopy(data, index + 1, data, index, numMoved);
        }
        data[--size] = null;
    }
    
    public int size() {
        return size;
    }
    
    private void ensureCapacity(int minCapacity) {
        if (minCapacity > data.length) {
            int newCapacity = data.length * 2;
            if (newCapacity < minCapacity) {
                newCapacity = minCapacity;
            }
            data = Arrays.copyOf(data, newCapacity);
        }
    }
    
    public static void main(String[] args) {
        MyArrayList<Integer> list = new MyArrayList<>();
        list.add(1);
        list.add(2);
        list.add(3);
        
        System.out.println("Element at index 1: " + list.get(1));
        
        list.remove(1);
        
        System.out.println("Size after removal: " + list.size());
        System.out.println("Element at index 1 after removal: " + list.get(1));
    }
}

总结

通过以上实现，我们成功地用纯数组实现了一个简化版的ArrayList，实现了基本的添加、获取、删除和动态扩容功能。尽管这只是一个基础版本，实际的ArrayList还包括更多复杂的功能和优化，如迭代器支持、并发安全等。理解这些基础原理对于深入理解Java集合框架及其性能是非常有帮助的。