ArrayList 扩容机制（源码笔记）

ArrayList 扩容机制

全局变量

 	//	如果使用无参构造器，初始化后第一次add时的默认数组长度
	private static final int DEFAULT_CAPACITY = 10;

    //	定义一个空数组以供使用
    private static final Object[] EMPTY_ELEMENTDATA = {};

    //	也是一个空数组，跟上边的空数组不同之处在于，这个是在默认构造器时返回的，扩容时需要用到这个作判断
    private static final Object[] DEFAULTCAPACITY_EMPTY_ELEMENTDATA = {};

    //	存放数组中的元素，注意此变量是transient修饰的，不参与序列化
    transient Object[] elementData; // non-private to simplify nested class access

    //	实际存放元素的个数，区别于elementData.length
    private int size;

构造函数

//	无参构造器，创建一个空数组，第一次add时的默认数组长度为DEFAULT_CAPACITY
public ArrayList() {
        this.elementData = DEFAULTCAPACITY_EMPTY_ELEMENTDATA;
}

//	参数为容器
public ArrayList(Collection<? extends E> c) {
        elementData = c.toArray();
        if ((size = elementData.length) != 0) {
            // c.toArray might (incorrectly) not return Object[] (see 6260652)
            if (elementData.getClass() != Object[].class)
                elementData = Arrays.copyOf(elementData, size, Object[].class);
        } else {
            // replace with empty array.
            this.elementData = EMPTY_ELEMENTDATA;
        }
    }

public ArrayList(int initialCapacity) {
        if (initialCapacity > 0) {
            //	创建长度为initialCapacity的数组
            this.elementData = new Object[initialCapacity];
        } else if (initialCapacity == 0) {
            this.elementData = EMPTY_ELEMENTDATA;
        } else {
            throw new IllegalArgumentException("Illegal Capacity: "+
                                               initialCapacity);
        }
    }

add()方法

//	参数为一个元素
public boolean add(E e) {
        ensureCapacityInternal(size + 1);  // Increments modCount!!
        elementData[size++] = e;
        return true;
    }

//	参数为下标和元素
public void add(int index, E element) {
        rangeCheckForAdd(index);

        ensureCapacityInternal(size + 1);  // Increments modCount!!
        System.arraycopy(elementData, index, elementData, index + 1,
                         size - index);
        elementData[index] = element;
        size++;
    }

计算出容量

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

//	计算要扩容到的最小容量
private static int calculateCapacity(Object[] elementData, int minCapacity) {
    //	先判断是否为无参构造的空数组，那么此时容量已经为 DEFAULT_CAPACITY
        if (elementData == DEFAULTCAPACITY_EMPTY_ELEMENTDATA) {
            return Math.max(DEFAULT_CAPACITY, minCapacity);
        }
        return minCapacity;
    }

//	如果此时数组的容量小于要扩容到的最小容量，那么就要执行扩容方法
private void ensureExplicitCapacity(int minCapacity) {
        modCount++;

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

扩容

//	将数组扩容到最小容量
private void grow(int minCapacity) {
        // 原容量（数组的长度）
        int oldCapacity = elementData.length;
    	// 新容量为原长度的1.5倍
        int newCapacity = oldCapacity + (oldCapacity >> 1);
    	//	如果新容量还不够大，那么就扩容到minCapacity
        if (newCapacity - minCapacity < 0)
            newCapacity = minCapacity;
    	//	防止整数溢出
        if (newCapacity - MAX_ARRAY_SIZE > 0)
            newCapacity = hugeCapacity(minCapacity);
        // 把原数组中的对象放入长度为newCapacity的新数组中
        elementData = Arrays.copyOf(elementData, newCapacity);
    }