深入理解ArrayList的动态扩容机制及应用

news2024/7/4 6:08:12

在java编程中，数据结构起着至关重要的作用，而ArrayList作为一种常用的动态数组，为我们在处理数据时提供了便利。其中，其独特的动态扩容机制更是为其赢得了广泛的应用。我们不管在工作还是面试中，都会遇到ArrayList，本文将深入探讨ArrayList的动态扩容机制，以便我们在工作或者面试中用到。

ArrayList简介

ArrayList是Java编程语言中的一个类，它实现了List接口，底层通过数组来存储元素。提供了对List 的添加（add）、删除（remove）、获取元素（get）等功能。其缺点是元素必须连续存储，所以增删慢，优点是查询快。ArrayList具有动态扩容的特性，这意味着它能够根据需要自动调整内部数组的大小，以适应不同数量的元素。

动态扩容详解

当我们向ArrayList 中添加元素(调用add()或者addAll())时,都调用了一个ensureCapacityInternal（）方法

我们来看源码：

add()

public boolean add(E e) {
    ensureCapacityInternal(size + 1);  // Increments modCount!!
    elementData[size++] = e;
    return true;
}

addAll()

public boolean addAll(Collection<? extends E> c) {
    Object[] a = c.toArray();
    int numNew = a.length;
    ensureCapacityInternal(size + numNew);  // Increments modCount
    System.arraycopy(a, 0, elementData, size, numNew);
    size += numNew;
    return numNew != 0;
}

从源码可以看到，这两个方法都调用了ensureCapacityInternal（）这个方法，参数是当前list的长度加上要插入的对象给个个数（单个对象的话为1，对象集合的话是集合的长度），既集合添加元素所需最小的长度

ensureCapacityInternal（）

private void ensureCapacityInternal(int minCapacity) {
    ensureExplicitCapacity(calculateCapacity(elementData, minCapacity));
}

ensureExplicitCapacity() 方法的入参是calculateCapacity（）（参数是存储实际元素的数组和集合添加元素所需最小的长度）方法处理后的值。

calculateCapacity（）

private static int calculateCapacity(Object[] elementData, int minCapacity) {
    if (elementData == DEFAULTCAPACITY_EMPTY_ELEMENTDATA) {
        return Math.max(DEFAULT_CAPACITY, minCapacity);
    }
    return minCapacity;
}

当前方法返回的值是如果源集合是空的，则返回默认容量（10）和元素集合添加元素所需最小的长度值比较，值大的一个，若不为空则返回minCapacity。

ensureExplicitCapacity()

private void ensureExplicitCapacity(int minCapacity) {
    modCount++;

    // overflow-conscious code
    if (minCapacity - elementData.length > 0)
        grow(minCapacity);
}

grow

private void grow(int minCapacity) {
    // overflow-conscious code
    int oldCapacity = elementData.length;
    int newCapacity = oldCapacity + (oldCapacity >> 1);
    if (newCapacity - minCapacity < 0)
        newCapacity = minCapacity;
    if (newCapacity - MAX_ARRAY_SIZE > 0)
        newCapacity = hugeCapacity(minCapacity);
    // minCapacity is usually close to size, so this is a win:
    elementData = Arrays.copyOf(elementData, newCapacity);
}