从源码看Java集合之ArrayList

Java集合之ArrayList - 吃透增删查改

从源码看初始化以及增删查改，学习ArrayList。

先来看下ArrayList定义的几个属性：

private static final int DEFAULT_CAPACITY = 10;
private static final Object[] EMPTY_ELEMENTDATA = {};
private static final Object[] DEFAULTCAPACITY_EMPTY_ELEMENTDATA = {};
transient Object[] elementData;  // 保存对象
private int size; // ArrayList的长度

从这里可以看到ArrayList内部使用数组实现的。

一. 初始化

1. ArrayList()

无参的构造器：

    /**
     * Constructs an empty list with an initial capacity of ten.
     */
    public ArrayList() {
        this.elementData = DEFAULTCAPACITY_EMPTY_ELEMENTDATA;
    }

可以看到这个构造器初始化了一个空数组。这里有个疑问，就是注释明明说是构造了一个长度为10的数组，其实这是在添加第一个元素的时候才进行的扩充。

2. ArrayList(int initialCapacity)

指定长度的构造器：

    /**
     * Constructs an empty list with the specified initial capacity.
     *
     * @param  initialCapacity  the initial capacity of the list
     * @throws IllegalArgumentException if the specified initial capacity
     *         is negative
     */
    public ArrayList(int initialCapacity) {
        if (initialCapacity > 0) {
            this.elementData = new Object[initialCapacity];
        } else if (initialCapacity == 0) {
            this.elementData = EMPTY_ELEMENTDATA;
        } else {
            throw new IllegalArgumentException("Illegal Capacity: "+
                                               initialCapacity);
        }
    }

这个构造器显式的指明了数组的长度，其实如果小于10的话，在添加第一个元素的时候还是会扩充到长度为10的数组。

二. 增加元素

1. add(E e)

    /**
     * Appends the specified element to the end of this list.
     *
     * @param e element to be appended to this list
     * @return <tt>true</tt> (as specified by {@link Collection#add})
     * 添加指定的元素到List的末尾
     */
    public boolean add(E e) {
        ensureCapacityInternal(size + 1);  // 扩充数组长度
        elementData[size++] = e;           // 最后一个赋值为e
        return true;
    }

那么它是如何扩充数组长度的呢？追踪代码来到grow(int minCapacity)方法:

/*
* 可以看到它是通过Arrays.copyOf方法将原数组拷贝到了一个数组长度为newCapacity的新数组里面
*/
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);
    }

2. add(int index, E element)

    /**
     * 在指定的位置添加一个元素
     */
    public void add(int index, E element) {
        rangeCheckForAdd(index); // 检查index是否合法

        ensureCapacityInternal(size + 1);  // 扩充数组长度
        System.arraycopy(elementData, index, elementData, index + 1,
                         size - index);  // 通过拷贝返回一个新数组
        elementData[index] = element;
        size++;
    }

这里用到了System类的arraycopy方法，它的用法如下：

System. arraycopy(Object src, int srcPos, Object dest, int destPos,int length)

src：原数组；

srcPos：源数组要复制的起始位置；

dest：目标数组；

destPos：目标数组放置的起始位置；

length：复制的长度

这个方法也可以实现自身的复制。上面的就是利用了自身赋值的特性进行数组的拷贝。相当于将index后面的所有元素后移了一位，将新元素插入到index位置。

三. 删除元素

1. remove(int index)

     /*
     * 和add(int index, E element)类似，使用System.arraycopy进行自身拷贝，相当于将index后面的元素
     * 像前移动了一位，覆盖掉需要删除的元素，将最后一个元素置位null，等待JVM自动回收
     */
    public E remove(int index) {
        rangeCheck(index);

        modCount++;
        E oldValue = elementData(index);

        int numMoved = size - index - 1;
        if (numMoved > 0)
            System.arraycopy(elementData, index+1, elementData, index,
                             numMoved);
        elementData[--size] = null; // clear to let GC do its work

        return oldValue;
    }

2. remove(Object o)

    /*
    * 先通过for循环找到o所在的位置，再通过fastRemove(index)移除，实现方法和remove(int index)一样
    */
    public boolean remove(Object o) {
        if (o == null) {
            for (int index = 0; index < size; index++)
                if (elementData[index] == null) {
                    fastRemove(index);
                    return true;
                }
        } else {
            for (int index = 0; index < size; index++)
                if (o.equals(elementData[index])) {
                    fastRemove(index);
                    return true;
                }
        }
        return false;
    }

四. 查找元素

    /*
    * 查找元素就比较简单了，直接通过数组的下角标进行返回
    */
    public E get(int index) {
        rangeCheck(index);

        return elementData(index);
    }

五. 修改元素

    /*
    * 修改元素也比较简单，直接通过数组的下角标进行赋值
    */
    public E set(int index, E element) {
        rangeCheck(index);

        E oldValue = elementData(index);
        elementData[index] = element;
        return oldValue;
    }

总结

从源码可以看到，ArrayList以数组方式实现，查找和修改的性能比较好，增加和删除的性能就比较差了。