ArrayList的扩容机制

ArrayList的扩容机制

默认构造函数，初始一个容量为10的数组elementData，但实际上这个数据在第一次加入后才回去申请10个内存空间，就是说第一次加入就会扩容为10。

下面看看第一次加入元素时，如何扩容的。

 public boolean add(E e) {
     //添加元素之前，先调用ensureCapacityInternal方法
     // 第一次加入时，size为0，size+1即为1
     ensureCapacityInternal(size + 1);  // Increments modCount!!
     //这里看到ArrayList添加元素的实质就相当于为数组赋值
     elementData[size++] = e;
     return true;
}

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

private static int calculateCapacity(Object[] elementData, int minCapacity) {
    if (elementData == DEFAULTCAPACITY_EMPTY_ELEMENTDATA) {
        // 第一次加入时，elementData为DEFAULTCAPACITY_EMPTY_ELEMENTDATA
        // DEFAULT_CAPACITY为10 minCapacity为1
        return Math.max(DEFAULT_CAPACITY, minCapacity);
    }
    return minCapacity;
}

private void ensureExplicitCapacity(int minCapacity) {
    modCount++;
    // 10 - 0= 9 > 0
    if (minCapacity - elementData.length > 0)
        // 执行grow进行扩容
        grow(minCapacity);
}

private void grow(int minCapacity) {
    // overflow-conscious code
    int oldCapacity = elementData.length;
    // 扩容为1.5倍，第一次的时候这里还是0，因为0的1.5倍还是0
    int newCapacity = oldCapacity + (oldCapacity >> 1);
    if (newCapacity - minCapacity < 0)
        // 第一次的时候 minCapacity为10 ，所以会进入这里，newCapacity就为10
        newCapacity = minCapacity;
    if (newCapacity - MAX_ARRAY_SIZE > 0)
        newCapacity = hugeCapacity(minCapacity);
    // minCapacity is usually close to size, so this is a win:
    // 将elementData的元素复制新数组并且长度为newCapacity
    elementData = Arrays.copyOf(elementData, newCapacity);
}

第二次扩容的时候，流程和上面一致。

ensureCapacity的作用

指定容量进行扩容

public void ensureCapacity(int minCapacity) {
        int minExpand = (elementData != DEFAULTCAPACITY_EMPTY_ELEMENTDATA)
            // any size if not default element table
            ? 0
            // larger than default for default empty table. It's already
            // supposed to be at default size.
            : DEFAULT_CAPACITY;

        if (minCapacity > minExpand) {
            ensureExplicitCapacity(minCapacity);
        }
    }

向 ArrayList 添加大量元素之前最好先使用ensureCapacity 方法，以减少扩容的次数。