迎难而上ArrayList，源码分析走一波

先看再点赞，给自己一点思考的时间，思考过后请毫不犹豫微信搜索【 沉默王二 】，关注这个长发飘飘却靠才华苟且的程序员。

本文 GitHub github.com/itwanger 已收录，里面还有技术大佬整理的面试题，以及二哥的系列文章。

关于 Java 基础、Java 面向对象编程、Java 字符串、Java 数组等方面的知识点已经可以告一段落了，小伙伴们可以在「沉默王二」公众号后台回复「 小白 」获取第二版手册。觉得不错的话，请随手转发给身边的小伙伴，赠人玫瑰，手有余香哈。

那么接下来，我开始肝 Java 集合方面的文章了，小伙伴们请默默为我鼓个掌，我能听得到，真的，别吝啬你的掌声，响起来。第一篇，必须得从 ArrayList 开始，毕竟 ArrayList 可以称得上是集合方面最常用的类了，估计没有之一。

ArrayList 实现了 List 接口，是基于数组实现的。小伙伴们都知道，数组的大小是固定的，创建的时候指定了大小，就不能再调整了，如果数组满了，就不能再添加任何元素了。ArrayList 是数组很好的替代方案，它提供了比数组更丰富的预定义方法（增删改查），并且大小是可以根据元素的多少进行自动调整的，非常灵活。

准备在 ArrayList 的第四个位置（下标为 3）上添加一个元素 55。

此时 ArrayList 中第五个位置以后的元素将会向后移动。

准备把 23 从 ArrayList 中移除。

此时下标为 7、8、9 的元素往前挪。

01、如何创建一个 ArrayList

ArrayList<String> alist = new ArrayList<String>();

可以通过上面的语句来创建一个字符串类型的 ArrayList（通过尖括号来限定 ArrayList 中元素的类型，如果尝试添加其他类型的元素，将会产生编译错误），更简化的写法如下：

List<String> alist = new ArrayList<>();

由于 ArrayList 实现了 List 接口，所以 alist 变量的类型可以是 List 类型；new 关键字声明后的尖括号中可以不再指定元素的类型，因为编译器可以通过前面尖括号中的类型进行智能推断。

如果非常确定 ArrayList 中元素的个数，在创建的时候还可以指定初始大小。

List<String> alist = new ArrayList<>(20);

这样做的好处是，可以有效地避免在添加新的元素时进行不必要的扩容。但通常情况下，我们很难确定  ArrayList 中元素的个数，因此一般不指定初始大小。

02、向 ArrayList 中添加一个元素

可以通过 add() 方法向 ArrayList 中添加一个元素，如果不指定下标的话，就默认添加在末尾。

alist.add("沉默王二");

感兴趣的小伙伴可以研究一下 add() 方法的源码，它在添加元素的时候会执行 grow() 方法进行扩容，这个是面试官特别喜欢考察的一个重点。

下面是 add(E e) 方法的源码：

public boolean add(E e) {
    modCount++;
    add(e, elementData, size);
    return true;
}

调用了私有的 add(E e, Object[] elementData, int s) 方法：

private void add(E e, Object[] elementData, int s) {
    if (s == elementData.length)
        elementData = grow();
    elementData[s] = e;
    size = s + 1;
}

然后调用了非常关键的 grow(int minCapacity) 方法：

private Object[] grow(int minCapacity) {
    int oldCapacity = elementData.length;
    if (oldCapacity > 0 || elementData != DEFAULTCAPACITY_EMPTY_ELEMENTDATA) {
        int newCapacity = ArraysSupport.newLength(oldCapacity,
                minCapacity - oldCapacity, /* minimum growth */
                oldCapacity >> 1           /* preferred growth */);
        return elementData = Arrays.copyOf(elementData, newCapacity);
    } else {
        return elementData = new Object[Math.max(DEFAULT_CAPACITY, minCapacity)];
    }
}

如果创建 ArrayList 的时候没有指定初始大小，那么 ArrayList 的初始大小就是 DEFAULT_CAPACITY：

private static final int DEFAULT_CAPACITY = 10;

可以容纳 10 个元素。

还可以通过 add(int index, E element) 方法把元素添加到指定的位置：

alist.add(0, "沉默王三");

add(int index, E element) 方法的源码如下：

public void add(int index, E element) {
    rangeCheckForAdd(index);
    modCount++;
    final int s;
    Object[] elementData;
    if ((s = size) == (elementData = this.elementData).length)
        elementData = grow();
    System.arraycopy(elementData, index,
            elementData, index + 1,
            s - index);
    elementData[index] = element;
    size = s + 1;
}

该方法会调用到一个非常重要的本地方法 System.arraycopy() ，它会对数组进行复制（要插入位置上的元素往后复制，参照文章一开头提到的两张图片）。

03、更新 ArrayList 中的元素

可以使用 set() 方法来更改 ArrayList 中的元素，需要提供下标和新元素。

alist.set(0, "沉默王四");

原来 0 位置上的元素为“沉默王三”，现在将其更新为“沉默王四”。

来看一下 set() 方法的源码：

public E set(int index, E element) {
    Objects.checkIndex(index, size);
    E oldValue = elementData(index);
    elementData[index] = element;
    return oldValue;
}

该方法会先对指定的下标进行检查，看是否越界，然后替换新值并返回旧值。

04、删除 ArrayList 中的元素

remove(int index) 方法用于删除指定下标位置上的元素， remove(Object o) 方法用于删除指定值的元素。

alist.remove(1);
alist.remove("沉默王四");

先来看 remove(int index) 方法的源码：

public E remove(int index) {
    Objects.checkIndex(index, size);
    final Object[] es = elementData;

    @SuppressWarnings("unchecked") E oldValue = (E) es[index];
    fastRemove(es, index);

    return oldValue;
}

该方法返回要删除的元素，真正的删除操作在 fastRemove(es, index) 方法中。

再来看 remove(Object o) 方法的源码：

public boolean remove(Object o) {
    final Object[] es = elementData;
    final int size = this.size;
    int i = 0;
    found: {
        if (o == null) {
            for (; i < size; i++)
                if (es[i] == null)
                    break found;
        } else {
            for (; i < size; i++)
                if (o.equals(es[i]))
                    break found;
        }
        return false;
    }
    fastRemove(es, i);
    return true;
}

该方法通过 break label 的方式找到要删除元素（null 的时候使用 == 操作符判断，非 null 的时候使用 equals() 方法，意味着如果有相同元素时，删除第一个）的下标，然后调用 fastRemove() 方法。

既然都调用了 fastRemove() 方法，那就继续来跟踪一下源码：

private void fastRemove(Object[] es, int i) {
    modCount++;
    final int newSize;
    if ((newSize = size - 1) > i)
        System.arraycopy(es, i + 1, es, i, newSize - i);
    es[size = newSize] = null;
}

当删除的是末尾的元素时，不需要复制数组，直接把末尾的元素赋值为 null 即可；否则的话，就需要调用 System.arraycopy() 对数组进行复制。参照文章一开头提到的第三张、第四张图片。

05、查找 ArrayList 中的元素

如果要正序查找一个元素，可以使用 indexOf() 方法；如果要倒序查找一个元素，可以使用 lastIndexOf() 方法。

alist.indexOf("沉默王二");
alist.lastIndexOf("沉默王二");

来看一下 indexOf() 方法的源码：

public int indexOf(Object o) {
    return indexOfRange(o, 0, size);
}

int indexOfRange(Object o, int start, int end) {
    Object[] es = elementData;
    if (o == null) {
        for (int i = start; i < end; i++) {
            if (es[i] == null) {
                return i;
            }
        }
    } else {
        for (int i = start; i < end; i++) {
            if (o.equals(es[i])) {
                return i;
            }
        }
    }
    return -1;
}

如果元素为 null 的时候使用“==”操作符，否则使用 equals() 方法——该方法不是 null 安全的。

lastIndexOf() 方法和 indexOf() 方法类似，不过遍历的时候从最后开始。

contains() 方法可以判断 ArrayList 中是否包含某个元素，其内部调用了 indexOf() 方法：

public boolean contains(Object o) {
    return indexOf(o) >= 0;
}

如果 ArrayList 中的元素是经过排序的，就可以使用二分查找法，效率更快。

Collections 类的 sort() 方法可以对 ArrayList 进行排序，该方法会按照字母顺序对 String 类型的列表进行排序。如果是自定义类型的列表，还可以指定 Comparator 进行排序。

List<String> copy = new ArrayList<>(alist);
copy.add("a");
copy.add("c");
copy.add("b");
copy.add("d");

Collections.sort(copy);
System.out.println(copy);

输出结果如下所示：

[a, b, c, d]

排序后就可以使用二分查找法了：

int index = Collections.binarySearch(copy, "b");

06、最后

关于 ArrayList，就先介绍这么多吧，通过源码的角度，我想小伙伴们一定对 ArrayList 有了更深刻的印象。

简单总结一下 ArrayList 的时间复杂度，方便后面学习 LinkedList 时作为一个对比。

1）通过下标（也就是 get(int index) ）访问一个元素的时间复杂度为 O(1)，因为是直达的，无论数据增大多少倍，耗时都不变。

2）添加一个元素（也就是 add() ）的时间复杂度为 O(1)，因为直接添加到末尾。

3）删除一个元素的时间复杂度为 O(n)，因为要遍历列表，数据量增大几倍，耗时也增大几倍。

4）查找一个未排序的列表时间复杂度为 O(n)，因为要遍历列表；查找排序过的列表时间复杂度为 O(log n)，因为可以使用二分查找法，当数据增大 n 倍时，耗时增大 logn 倍（这里的 log 是以 2 为底的，每找一次排除一半的可能）。

我是沉默王二，一枚有颜值却靠才华苟且的程序员。 关注即可提升学习效率，别忘了三连啊，点赞、收藏、留言，我不挑，奥利给 。

注：如果文章有任何问题，欢迎毫不留情地指正。

很多读者都同情我说，“二哥，你像母猪似的日更原创累不累？”我只能说写作不易，且行且珍惜啊，关键是我真的喜欢写作。最后，欢迎微信搜索「 沉默王二 」第一时间阅读，回复「 简历 」更有阿里大佬的简历模板，本文 GitHub github.com/itwanger 已收录，欢迎阅读原文 star。