介绍

ArrayList是一个动态可调整大小的数组列表。内部是基于数组实现的,大小会自动增长,支持随机访问。

  1. public class ArrayList<E> extends AbstractList<E>
  2. implements List<E>, RandomAccess, Cloneable, java.io.Serializable

类图如下:
image.png

数据结构

静态常量

可以看到数组的大小默认为10

  1. /**
  2. * Default initial capacity.
  3. */
  4. private static final int DEFAULT_CAPACITY = 10;
  5. /**
  6. * Shared empty array instance used for empty instances.
  7. */
  8. private static final Object[] EMPTY_ELEMENTDATA = {};
  9. /**
  10. * Shared empty array instance used for default sized empty instances. We
  11. * distinguish this from EMPTY_ELEMENTDATA to know how much to inflate when
  12. * first element is added.
  13. */
  14. private static final Object[] DEFAULTCAPACITY_EMPTY_ELEMENTDATA = {};
  15. /**
  16. * 分配的数组的最大大小。 尝试分配更大的数组可能会导致 OutOfMemoryError
  17. *
  18. */
  19. private static final int MAX_ARRAY_SIZE = Integer.MAX_VALUE - 8;

成员变量

elementData存储了实际的元素,transient修饰符保证他不被序列化,size是数组中元素的个数。

  1. /**
  2. * The array buffer into which the elements of the ArrayList are stored.
  3. * The capacity of the ArrayList is the length of this array buffer. Any
  4. * empty ArrayList with elementData == DEFAULTCAPACITY_EMPTY_ELEMENTDATA
  5. * will be expanded to DEFAULT_CAPACITY when the first element is added.
  6. */
  7. transient Object[] elementData; // non-private to simplify nested class access
  8. /**
  9. * The size of the ArrayList (the number of elements it contains).
  10. *
  11. * @serial
  12. */
  13. private int size;

构造方法

构造方法根据指定的大小初始化elementData数组,注意传0和不传是不一样的,
传0会初始化为EMPTY_ELEMENTDATA,不传初始化为DEFAULTCAPACITY_EMPTY_ELEMENTDATA

  1. public ArrayList(int initialCapacity) {
  2. if (initialCapacity > 0) {
  3. this.elementData = new Object[initialCapacity];
  4. } else if (initialCapacity == 0) {
  5. this.elementData = EMPTY_ELEMENTDATA;
  6. } else {
  7. throw new IllegalArgumentException("Illegal Capacity: "+
  8. initialCapacity);
  9. }
  10. }
  11. public ArrayList() {
  12. this.elementData = DEFAULTCAPACITY_EMPTY_ELEMENTDATA;
  13. }

添加元素 & 扩容 add(E e)

添加元素时使用 ensureCapacityInternal() 方法来保证容量足够,如果不够时,需要使用 grow() 方法进行扩容,新容量的大小为 oldCapacity + (oldCapacity >> 1),即 oldCapacity+oldCapacity/2。其中 oldCapacity >> 1 需要取整,所以新容量大约是旧容量的 1.5 倍左右。(oldCapacity 为偶数就是 1.5 倍,为奇数就是 1.5 倍 - 0.5 5 + 2.5
扩容操作需要调用 Arrays.copyOf() 把原数组整个复制到新数组中,这个操作代价很高,因此最好在创建 ArrayList 对象时就指定大概的容量大小,减少扩容操作的次数。

  1. public boolean add(E e) {
  2. // 保证容量大小必须大于等于size + 1
  3. ensureCapacityInternal(size + 1); // Increments modCount!!
  4. // 赋值
  5. elementData[size++] = e;
  6. return true;
  7. }
  8. private void ensureCapacityInternal(int minCapacity) {
  9. // 数组为空,minCapacity从DEFAULT_CAPACITY(10)和minCapacity取最大值
  10. if (elementData == DEFAULTCAPACITY_EMPTY_ELEMENTDATA) {
  11. minCapacity = Math.max(DEFAULT_CAPACITY, minCapacity);
  12. }
  13. // 保证容量大小至少有minCapacity个
  14. ensureExplicitCapacity(minCapacity);
  15. }
  16. private void ensureExplicitCapacity(int minCapacity) {
  17. modCount++;
  18. // overflow-conscious code
  19. // 最小容量 - 当前数组长度 > 0,说明当前数组大小不够,需要扩容,调用grow函数
  20. if (minCapacity - elementData.length > 0)
  21. grow(minCapacity);
  22. }
  23. // 扩容
  24. private void grow(int minCapacity) {
  25. // overflow-conscious code
  26. // 旧数组的长度
  27. int oldCapacity = elementData.length;
  28. // 新数组长度为旧数组的长度 * 1.5
  29. // 如果旧数组是奇数,最低位会丢失,等同于oldCapacity + (oldCapacity-1) >> 1
  30. int newCapacity = oldCapacity + (oldCapacity >> 1);
  31. // 判断newCapacity < minCapacity,说明扩容一次后容量达不到minCapacity
  32. if (newCapacity - minCapacity < 0)
  33. // newCapacity赋值为minCapacity
  34. newCapacity = minCapacity;
  35. // 超过最大分配数组大小
  36. if (newCapacity - MAX_ARRAY_SIZE > 0)
  37. newCapacity = hugeCapacity(minCapacity);
  38. // minCapacity is usually close to size, so this is a win:
  39. // 数组所有元素复制到新数组中,并赋值给elementData
  40. elementData = Arrays.copyOf(elementData, newCapacity);
  41. }
  42. private static int hugeCapacity(int minCapacity) {
  43. if (minCapacity < 0) // overflow
  44. throw new OutOfMemoryError();
  45. // 如果超过MAX_ARRAY_SIZE 赋值为Integer.MAX_VALUE
  46. return (minCapacity > MAX_ARRAY_SIZE) ?
  47. Integer.MAX_VALUE :
  48. MAX_ARRAY_SIZE;
  49. }
  50. Arrays.copyOf:
  51. public static <T> T[] copyOf(T[] original, int newLength) {
  52. // 获取数组的类型,复制指定长度的元素
  53. return (T[]) copyOf(original, newLength, original.getClass());
  54. }
  55. public static <T,U> T[] copyOf(U[] original, int newLength,
  56. Class<? extends T[]> newType) {
  57. @SuppressWarnings("unchecked")
  58. // 创建一个空的指定类型的数组,
  59. T[] copy = ((Object)newType == (Object)Object[].class)
  60. ? (T[]) new Object[newLength]
  61. : (T[]) Array.newInstance(newType.getComponentType(), newLength);
  62. // 从original数值的0下标开始复制到copy数组的0下标处进行复制,复制的个数是newLength
  63. System.arraycopy(original, 0, copy, 0,
  64. Math.min(original.length, newLength));
  65. return copy;
  66. }

通过下标获取元素 get(int index)

  1. public E get(int index) {
  2. // 下标范围检查
  3. rangeCheck(index);
  4. // 通过数组下标直接获取元素
  5. return elementData(index);
  6. }
  7. private void rangeCheck(int index) {
  8. // 下标超过size,会有自定义提示,Index: {index}, Size: {size}
  9. // 如果下标是负数,数组本身也会抛出IndexOutOfBoundsException
  10. if (index >= size)
  11. throw new IndexOutOfBoundsException(outOfBoundsMsg(index));
  12. }
  13. E elementData(int index) {
  14. return (E) elementData[index];
  15. }

根据index移除元素 remove(int index)

  1. // 根据下标移除元素
  2. public E remove(int index) {
  3. // 下标范围检查
  4. rangeCheck(index);
  5. modCount++;
  6. // 下标取值
  7. E oldValue = elementData(index);
  8. // 移动元素的数量,index后面的元素的个数
  9. // 因为index从0开始,所有index后面元素数量应该是size - (index + 1) = size -index -1
  10. int numMoved = size - index - 1;
  11. if (numMoved > 0)
  12. // 下标从index之后的所有元素全部往前移动一位
  13. System.arraycopy(elementData, index+1, elementData, index,
  14. numMoved);
  15. // 移除末尾下标位置元素
  16. elementData[--size] = null; // clear to let GC do its work
  17. // 返回目标的旧值
  18. return oldValue;
  19. }
  20. private void rangeCheck(int index) {
  21. if (index >= size)
  22. throw new IndexOutOfBoundsException(outOfBoundsMsg(index));
  23. }
  24. @SuppressWarnings("unchecked")
  25. E elementData(int index) {
  26. return (E) elementData[index];
  27. }
  28. System.arraycopy
  29. /**
  30. * src – 源数组。
  31. * srcPos – 源数组中的起始位置。
  32. * dest – 目标数组。
  33. * destPos – 目标数据中的起始位置。
  34. * length – 要复制的数组元素的数量。
  35. */
  36. public static native void arraycopy(Object src, int srcPos,
  37. Object dest, int destPos,
  38. int length);

根据object移除元素 remove(Object o)

  1. public boolean remove(Object o) {
  2. // 元素为null
  3. if (o == null) {
  4. for (int index = 0; index < size; index++)
  5. // 遍历移除第一个值为null的元素
  6. if (elementData[index] == null) {
  7. fastRemove(index);
  8. return true;
  9. }
  10. } else {
  11. for (int index = 0; index < size; index++)
  12. // 遍历移除第一个equals相等的元素
  13. if (o.equals(elementData[index])) {
  14. fastRemove(index);
  15. return true;
  16. }
  17. }
  18. return false;
  19. }
  20. /*
  21. * Private remove method that skips bounds checking and does not
  22. * return the value removed.
  23. */
  24. private void fastRemove(int index) {
  25. modCount++;
  26. // remove后需要移动的元素个数
  27. int numMoved = size - index - 1;
  28. if (numMoved > 0)
  29. // index之后的所有元素向前移动一位
  30. System.arraycopy(elementData, index+1, elementData, index,
  31. numMoved);
  32. // 末尾下标元素赋值为null
  33. elementData[--size] = null; // clear to let GC do its work
  34. }

这里只有一个关键点,remove(Object o)方法只会按顺序移除第一个equals的元素,如果equals有匹配多个元素话,则只会移除第一个。