ArrayList最基本的用法:
import java.util.ArrayList; public class ArrayListDemo { public static void main(String[] args) { ArrayList<String> list = new ArrayList<>(); list.add("demo"); String s = list.get(0); list.remove(0); } }
总共分为4步
声明一个ArrayList
进入ArrayList的构造函数
public class ArrayList<E> extends AbstractList<E> implements List<E>, RandomAccess, Cloneable, java.io.Serializable{ // ... transient Object[] elementData; // ... private static final Object[] DEFAULTCAPACITY_EMPTY_ELEMENTDATA = {}; // ... public ArrayList() { this.elementData = DEFAULTCAPACITY_EMPTY_ELEMENTDATA; } // ... }
elementData
为实际上 ArrayList 存储数据的 Object 数组,DEFAULTCAPACITY_EMPTY_ELEMENTDATA
为一个空数组,所以初始化的结果就是将一个空数组赋值给了实际存储数组的数组,其他的什么也没有做
添加一个元素
public boolean add(E e) { modCount++; add(e, elementData, size); return true; }
其中 modeCount
的作用是记录对这个 list 的修改次数,包括 add/delete 等操作。通过注释可以发现,这个值的作用是用在迭代器中,如果在迭代的过程中修改了数组,则抛出 ConcurrentModificationException 这个异常。
public static void main(String[] args) { ArrayList<String> list = new ArrayList<>(); list.add("demo1"); list.add("demo2"); list.add("demo3"); Iterator<String> iterator = list.iterator(); while (iterator.hasNext()) { String next = iterator.next(); System.out.println(next); if (next.equals("demo2")) { list.remove(next); } } }
进入最主要的 add 函数,发现接受 3 个参数,当前需要添加的值,目前已经存的数据,当前的长度
private void add(E e, Object[] elementData, int s) { if (s == elementData.length) elementData = grow(); elementData[s] = e; size = s + 1; }
其中首先判断当前的 elementData.length
和 list.size
的大小,如果两个值相等,则意味着 elementData
已经被填满,需要对 elementData
数组进行扩容,否则直接可以在 elementData
中的第 size 位添加元素,并把 size 加1即可
首次添加元素,通过上面的构造函数可知,此时的 elementData
为空数组,且当前 size 为0,满足条件,需要对 elementData
数组进行扩容,进入 grow()
。
private Object[] grow() { return grow(size + 1); } 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)]; } }
参数 size + 1表示此时需要扩容的最小容量 minCapacity
如果是第一次扩容,进入 else 分支,可以看到是初始化 DEFAULT_CAPACITY
和 minCapacity
的最大值,其中 DEFAULT_CAPACITY
的值是10 ,于是 ArrayList 的初始化容量为 10 就是从这里得出
如果不是第一次扩容,则进入 if 分支,可以看到新的容量是通过 ArraysSupport.newLength()
得出,进入对应的函数。
public static int newLength(int oldLength, int minGrowth, int prefGrowth) { // assert oldLength >= 0 // assert minGrowth > 0 int newLength = Math.max(minGrowth, prefGrowth) + oldLength; if (newLength - MAX_ARRAY_LENGTH <= 0) { return newLength; } return hugeLength(oldLength, minGrowth); }
可以看到该函数接受三个参数 oldLength
当前长度,minGrowth
最小扩容长度,prefGrowth
最佳长度,此时最佳长度为 oldCapacity >> 1
,即当前长度的一半。
可以看到通过这个函数,返回的新长度为 prefGrowth
和 minGrowth
的最大值再加上当前长度,即当前长度的1.5倍,即 ArrayList 默认的 1.5 倍扩容来源于此。
在hugeLength()
中,对于特别大的扩容数量做了单独处理,即 Integer.MAX_VALUE - 8
由此 ArrayList 的 add 流程就基本结束。
获取一个元素
public E get(int index) { Objects.checkIndex(index, size); return elementData(index); }
首先检查传入的 index
是否超过 size
然后进入 elementData(index)
函数
E elementData(int index) { return (E) elementData[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; }
首先同样的,需要校验传入 index
是否超过 size
然后通过下标将需要删除的元素取出
然后进入 fastRemove(es, index)
将该元素删除,进入对应函数
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; }
与 add 同样的,首先需要对 modCount
进行 +1
其中 (newSize = size - 1) > i
的作用是判断当前的删除元素是否为最后一个,如果是的话,则需要将该元素的后面所有元素前移一位
最后将最后一位置 null