Java集合ArrayList源码解读
最近在回顾数据结构,想到JDK这样好的代码资源不利用有点可惜,这是第一篇,花了心思。篇幅有点长,希望想看的朋友认真看下去,提出宝贵的意见。 :)
类继承体系图
内部原理
ArrayList 的3个字段
private transient Object[] elementData; //对象数组,用于存储 持有对象的 引用 private int size; //代表了 ArrayList 的长度。随着插入 删除 添加 而改变。 protected transient int modCount = 0; //从AbstractList继承得到,这个字段最后介绍,先忽视它。
elementData 是一个Object 数组,这是为了兼容任何类型(Java泛型是所有实际类型共享一份代码模板的!!!)。数组保存的实质是持有对象的引用(reference)。引用又可以理解为 对象的“遥控器”(如下图)。
所以要强调的是:对于保存引用类型,java中的数组,和各种集合类,持有的是对象的引用,而不是对象本身。
构造函数
空ArrayList 的优化 : new 一个空的ArrayList是很常见的做法,为了不让每个空ArrayList都创建一个空数组实例,ArrayList内部有一个用于共享,静态的空数组对象。当创建空的ArrayList时,内部的 elementData 就会指向这个共享的空数组: EMPTY_ELEMENTDATA
同时也会发现:当new 一个空ArrayList 时,不会马上在内存中分配 DEFAULT_CAPACITY =10 个长度的elementData数组,而是等到第一个元素被添加时,才会去申请分配默认长度 为10 的elementData 数组。
private static final Object[] EMPTY_ELEMENTDATA = {}; //用于所有空ArrayList共享的 空数组,注意它是静态字段。 private static final int DEFAULT_CAPACITY = 10; //第一次分配容量时,默认的初始容量大小 ////////////////////////////////////////////// public ArrayList() { super(); this.elementData = EMPTY_ELEMENTDATA; //指向空数组 }
当然,如果在new 时指定了 初始容量initCapacity,就会立刻 创建一个 长度为 initialCapacity 大小的 elementData 数组
public ArrayList(int initialCapacity) { super(); if (initialCapacity < 0) throw new IllegalArgumentException("Illegal Capacity: "+initialCapacity); this.elementData = new Object[initialCapacity];
}
public ArrayList(Collection<? extends E> c) { elementData = c.toArray(); size = elementData.length; // c.toArray might (incorrectly) not return Object[] (see 6260652) if (elementData.getClass() != Object[].class) elementData = Arrays.copyOf(elementData, size, Object[].class); }
容量增长策略
虽然ArrayList 号称 动态数组,其长度自动调整,但是内部 保存持有对象的引用的数组elementData却是一个普通的数组(Java数组的长度是不可变的),因此,ArrayList就必须实现容量扩展操作,在elementData 满 时,让elementData指向长度更大的数组。
以保证在任何时候:elementData.length >= size
elementData 的长度就 是 ArrayList 的容量(capacity)。
以下是关于容量调配的API(绿色标记的是public方法,红色的是private)
查看源码时,我发现,向ArrayList 中添加新元素的时,都会先 使用 ensureCapacityInternal 这个函数去检查容量是否够用。如图中,参数为 size+1 ,也就是要保证容量至少为size+1个,这样新元素才能放得下。我们去看看这个函数的实现。
/* 函数作用:用于确保 ArrayList 的容量至少有 参数指定的minCapacity个 */ private void ensureCapacityInternal(int minCapacity) { if (elementData == EMPTY_ELEMENTDATA) { //这个if用于处理 往 空ArrayList中添加第一个元素时的情况 minCapacity = Math.max(DEFAULT_CAPACITY, minCapacity); } ensureExplicitCapacity(minCapacity); //内部又委托了这个函数,继续往下看 }
/* 函数作用:用于确保 ArrayList 的容量至少有 参数指定的minCapacity个 */ private void ensureExplicitCapacity(int minCapacity) { modCount++; // overflow-conscious code if (minCapacity - elementData.length > 0) //如果当前容量 真的比 要求的 minCapacity 小 grow(minCapacity); //那就交给grow函数去扩张容量去吧。 }
最终的增长逻辑都是委托grow() 方法实现的。只要调用了grow这个函数,就一定会使容量增加。我们来看看容量到底是怎么增加的。
private void grow(int minCapacity) { // overflow-conscious code int oldCapacity = elementData.length; int newCapacity = oldCapacity + (oldCapacity >> 1); //先让 newCapacity = 原来的1.5倍,这是默认的增长策略。 //再接着来判断这个策略是否合适 if (newCapacity - minCapacity < 0) //如果变为原来的1.5倍还是不够 newCapacity = minCapacity; //那就要多少给多少吧。 if (newCapacity - MAX_ARRAY_SIZE > 0) //如果要求的容量超大,(大于MAX_ARRAY_SIZE),这个时候就必须慎重。怎么办? newCapacity = hugeCapacity(minCapacity); //交给 hugeCapacity函数去决定到底给多少,这个函数会返回 MAX_ARRAY_SIZE 或者 Integer.MAX_VALUE。 elementData = Arrays.copyOf(elementData, newCapacity); //经过一轮轮的判断,终于定下了新的容量,好,调整elementData的容量为 newCapacity! }
注:newCapacit 为需求容量,oldCapacity原来的容量
1、oldCapacity < newCapacit <= oldCapacity * 1.5
平缓增长的正常情况。例如原来容量为10 ,现在期望达到11个,则会扩展 调整到 10 + (10 >> 1) =15个 ,也就是变为 原来的1.5倍。
2、oldCapacity*1.5 < newCapacit <= MAX_ARRAY_SIZE
默认的1.5倍增长还不够,期望的容量有点大。例如原来容量为10 ,现在请求100个,那么就会实打实的扩展到 100 个的容量。
3、MAX_ARRAY_SIZE < newCapacit <= Interger.MAX_VALUE
极端情况,期望的容量超大 。在这个情况下,newCapacity的值最终是由hugeCapacity函数决定的,它会返回 MAX_ARRAY_SIZE 或者 Integer.MAX_VALUE
//有些Java虚拟机会在数组内存的头部保留 一些字节的容量来存储这个内存块的相关信息,因此,数组的 安全最大长度是 : Integer.MAX_VALUE - 8 private static final int MAX_ARRAY_SIZE = Integer.MAX_VALUE - 8;
public void trimToSize() { modCount++; if (size < elementData.length) { elementData = Arrays.copyOf(elementData, size); } }
公开接口:一次性决定ArrayList 的最大容量,避免半路上不断的调整容量带来开销(如果你真的很确定你需要的ArrayList 的最大长度,否则没必要)
public void ensureCapacity(int minCapacity) { int minExpand = (elementData != EMPTY_ELEMENTDATA) // any size if real element table ? 0 // larger than default for empty table. It's already supposed to be // at default size. : DEFAULT_CAPACITY; if (minCapacity > minExpand) { ensureExplicitCapacity(minCapacity); } }
顺序表的短板:插入和删除
先看一道试题:
在java中,下面哪一个复制数组的方法最高效?
A、循环赋值B、Arrays.copyOfC、System.arraycopyD、clone正确答案: C
public static native void arraycopy(Object src, int srcPos, Object dest, int destPos, int length); /* 从数组src中拷贝元素到 dest中 。我们把src数组叫做源数组,把dest数组叫做目的数组 src 源数组的引用 srcPos 源数组的拷贝起始索引 dest 目的数组的引用 destPos 目的数组的拷贝起始索引 length 有length个元素会被拷贝 如果参数 src 和 dest 引用相同的数组对象(注意,它允许内存重叠),则复制的执行过程就好像首先将 srcPos 到 srcPos+length-1 位置的组件复制到 一个带有 length 组件的临时数组,然后再将此临时数组的内容复制到目标数组的 destPos 到 destPos+length-1 位置一样。 这就好像是C语言标准库中的memmove 使用注意: 1、src 和 dest不能为null,否则会抛出NullPointerExcaption异常 2、src和 dest 的类型要兼容或者一样。 3、你必须精确把握好起始索引和拷贝长度的关系,有差错就是抛异常。 */
ArrayList在删除和插入 移动数组元素时,就是使用的这个API。我们来看看。
插入一个元素到指定位置
public void add(int index, E element) { rangeCheckForAdd(index); //检查index 的合法性:逻辑如下 ,后面不再列出 /* if (index > size || index < 0) throw new IndexOutOfBoundsException(outOfBoundsMsg(index)); */ ensureCapacityInternal(size + 1); // 确保容量足够 //elementData 的元素的移动,让出空间 System.arraycopy(elementData, index, elementData, index + 1,size - index); elementData[index] = element; //插入新的元素的引用 size++; }
插入一个集合的元素到指定的位置
public boolean addAll(int index, Collection<? extends E> c) { rangeCheckForAdd(index); //index 合法性检查 Object[] a = c.toArray(); //先转化为数组 int numNew = a.length; ensureCapacityInternal(size + numNew); // 确保容量足够 int numMoved = size - index; if (numMoved > 0) //原来的数组让出空间 System.arraycopy(elementData, index, elementData, index + numNew, numMoved); //将新的元素拷贝到腾出的空间去。 System.arraycopy(a, 0, elementData, index, numNew); size += numNew; return numNew != 0; }
画个图打开你的脑洞(插入2个元素的集合到ArrayList 的开头)
删除元素
/* 清空ArrayList中的所有持有对象。让size为0. 实质操作是让保存持有对象的引用的数组elementData 的元素全部为null。 所以我们会试想:如果在清空操作后 ,没有其它引用指向ArrayList原来持有的对象了,那么这个对象就会等待被GC 回收。 例子: ArrayList<Person> arr = new ArrayList<Person>(); Person p1 = new Person(); arr.add(p1); //添加第一个持有对象 arr.add(new Person()); //添加第二个持有对象,匿名的。 arr.clear(); //第一个持有对象因为还有引用p1 的指向,所以暂时不会等待GC回收 //第二个对象由于没有引用指向。则会等待GC回收。 */ public void clear() { modCount++; // clear to let GC do its work for (int i = 0; i < size; i++) elementData[i] = null; size = 0; }
按对象删除:删除第一个找到的元素,立刻返回。如果没找到,则什么也不做,返回false 。如果o不为null,则调用equals方法去查找。时间复杂度为O(n)。
public boolean remove(Object o) { if (o == null) //如果要删除null { for (int index = 0; index < size; index++) if (elementData[index] == null) //从前往后遍历查找,一旦找到为null ,删除之,返回 { fastRemove(index); //fastRemove 是对 System.arraycopy 的封装 return true; } } else //如果要删除某个真实存在的对象 { for (int index = 0; index < size; index++) if (o.equals(elementData[index])) //从前往后遍历查找,一旦找到,删除之,返回 { fastRemove(index); //fastRemove 是对 System.arraycopy 的封装 return true; } } return false; }
按索引删除:删除指定索引上的元素。主要耗时是在System.arraycopy 的执行上 。
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; }
只保留 ArrayList中 和 集合c 相同元素,其它的都删除。相同是指调用equals返回 true (或者都是null)
public boolean retainAll(Collection<?> c) { return batchRemove(c, true); }
删除 ArrayList中 和 集合c 相同元素。相同是指调用equals返回 true (或者都是null)
public boolean removeAll(Collection<?> c) { return batchRemove(c, false); }
追加元素
追加一个元素到末尾
public boolean add(E e) { ensureCapacityInternal(size + 1); // 确保容量足够 elementData[size++] = e; return true; }
追加一个集合的元素到末尾
public boolean addAll(Collection<? extends E> c) { Object[] a = c.toArray(); int numNew = a.length; ensureCapacityInternal(size + numNew); // Increments modCount System.arraycopy(a, 0, elementData, size, numNew); size += numNew; return numNew != 0; }
GET / SET
//对数组访问的简单封装。方便访问,避免重复的cast操作。是默认的包访问权限 @SuppressWarnings("unchecked") E elementData(int index) { return (E) elementData[index]; }
//获取某个位置上的元素。 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中是否包含某个对象 public boolean contains(Object o) { return indexOf(o) >= 0; } //返回第一个找到的对象的索引。找不到则返回-1 public int indexOf(Object o) { if (o == null) { //如果要找null for (int i = 0; i < size; i++) if (elementData[i]==null) return i; } else { //如果要找真实的对象 for (int i = 0; i < size; i++) if (o.equals(elementData[i])) //调用equals return i; } return -1; } //同上,只不过是从后往前查找。 public int lastIndexOf(Object o) { if (o == null) { for (int i = size-1; i >= 0; i--) if (elementData[i]==null) return i; } else { for (int i = size-1; i >= 0; i--) if (o.equals(elementData[i])) return i; } return -1; }
与数组的转换
转化为数组,是指将ArrayList中的elementData 保存的对象引用全部拷贝到一个新的数组中去,而对象并没有拷贝。
提供了下面2个函数
public Object[] toArray() { return Arrays.copyOf(elementData, size); } @SuppressWarnings("unchecked") public <T> T[] toArray(T[] a) { if (a.length < size) return (T[]) Arrays.copyOf(elementData, size, a.getClass()); System.arraycopy(elementData, 0, a, 0, size); if (a.length > size) a[size] = null; return a; }
第一个用的很少,因为它返回的是Object数组。也就是 和 elementData 一模一样的一份拷贝。而我们往往需要的是实际类型的数组。
你可能会写下面这样的代码
ArrayList<String> arr = new ArrayList<>(); arr.add("abc"); String[] ss = (String[])arr.toArray(); //不可以,运行时异常!
在Java中,只有相互兼容,有继承关系,满足里氏转化原则的才能 转型(向下或者向上)
然而,Object [] 和String[]没有继承关系,而是处于同一层次的类型。这和他们保存的类型是否有继承关系无关。同样的道理,扩展到泛型中去,ArrayList<Object> 和 ArrayList<String> 也是不能转化的,不兼容的,因为类型擦除,这两个类型实际是相同的,都是ArrayList。
因此,一般会使用第二个泛型方法
String [] ss = list.toArray(new String[0])
迭代器
关于迭代器的特点,用法,接口等这里就不详细介绍了,我直接进入主题。
为什么迭代过程中对ArrayList 的修改会抛异常?哪些是运行的?哪些不允许?
private static void bad() { ArrayList<Integer> arr = new ArrayList<>(); arr.add(0); arr.add(1); arr.add(2); Iterator<Integer> iter = arr.iterator(); while(iter.hasNext()) { Integer ele = iter.next(); //取得当前迭代出的这个元素 if(ele.intValue() /2 !=0) //如果这个元素是奇数 arr.remove(ele); //则从表中删除它 } /*Exception in thread "main" java.util.ConcurrentModificationException * * * */
}
先放下问题,来研究一下modCount 和 Itr 这个成员内部类(ArrayList 的迭代器)
modCount 的值就是代表list 发生结构性修改的次数。 什么叫结构性修改呢?比如插入,删除,追加等导致 size改变 和搅乱ArrayList 的结构的操作。 这个值主要用在ArrayLsit的迭代器中,用于检查迭代过程中表本身是否被意外的结构性修改。如果是,就会抛出 ConcurrentModificationException异常。 protected transient int modCount = 0; //从AbstractList继承得到
下面的方法都会使得modCount 加1 ,还有其它的方法也会。。。
ArrayLsit 的 迭代器 Itr
1 private class Itr implements Iterator<E> 2 { 3 int cursor; // 下一个将被迭代出的元素的index 4 int lastRet = -1; // 上一次被跌带出去的元素的index,如果是-1,则不合法。 5 int expectedModCount = modCount; // 初始化为Arraylist对象的字段: modCount 6 7 public boolean hasNext() { //迭代终点检查器,如果返回true则可以继续迭代 8 return cursor != size; 9 } 10 11 //在构造迭代器对象时,expectedModCount 的值被初始化为最开始的modCount。 12 //这个函数的目的就是:检查从迭代前,到跌倒结束过程中,ArrayList 的 modCount 的值是否一直和迭代器的字段exceptedModeCount一样 13 final void checkForComodification() 14 { 15 if (modCount != expectedModCount) 16 throw new ConcurrentModificationException(); 17 } 18 19 public E next() { /*返回下一个将要被迭代出的元素*/ 20 checkForComodification(); //每次调用next 都会检查 modCount 和 expectedModCount是否一致 21 int i = cursor; 22 23 if (i >= size) 24 throw new NoSuchElementException(); 25 Object[] elementData = ArrayList.this.elementData; 26 if (i >= elementData.length) 27 throw new ConcurrentModificationException(); 28 29 cursor = i + 1; 30 return (E) elementData[lastRet = i]; //返回迭代出的元素,并且让lastRet为已经被迭代出的元素的index 31 } 32 33 public void remove() { /*删除迭代宿主 的 最近的一个已经 被迭代出的元素*/ 34 if (lastRet < 0) //这个判断是为了限制:必须先把要删除的元素通过next迭代出来,才能再调用remove删除它 35 throw new IllegalStateException(); //请看第43行,又把它重置为 -1 。只有next的调用才会把它赋值为合法的index。(30行) 36 37 38 checkForComodification(); 39 40 try { 41 ArrayList.this.remove(lastRet); //直接调用了 宿主的remove 方法删除元素 42 cursor = lastRet; 43 lastRet = -1; 44 expectedModCount = modCount; //再让exceptedModCount 和modCount 保持一致,这样就避免了next()抛异常 45 46 } catch (IndexOutOfBoundsException ex) { 47 throw new ConcurrentModificationException(); 48 } 49 } 50 51 52 53 }
可以发现,抛出异常的根本是: modCount != expectedModCount 。 而迭代器构造时,expectedModCount= modCount,所以在迭代过程中,如果我们不使用ArrayList的改变modCount 的方法,就不会抛出异常。难道我们就必须这样小心翼翼吗?不,如果要删除ArrayList 的元素,我们可以使用迭代器的remove方法。如下:
private static void ok() { ArrayList<Integer> arr = new ArrayList<>(); arr.add(0); arr.add(1); arr.add(2); Iterator<Integer> iter = arr.iterator(); while(iter.hasNext()) { if(iter.next().intValue() /2 !=0) //如果是奇数 iter.remove(); //通过迭代器把它删除是OK的 } System.out.println(arr); }
同样,Java 5 引入的foreach 也是如此,因为foreach只不过是对迭代器的语法包装。编译后,实质还是调用了迭代器的方法。
for(Integer i : arr) { if(i.intValue() / 2!=0) arr.remove(i); //同样抛异常,foreach本质就是使用迭代器 }
其它
/*获得ArrayList 的size */ public int size()
{ return size; } /*获得ArrayList 是否为空 */ public boolean isEmpty()
{ return size == 0; } //clone public Object clone() { try { @SuppressWarnings("unchecked") ArrayList<E> v = (ArrayList<E>) super.clone(); v.elementData = Arrays.copyOf(elementData, size); v.modCount = 0; return v; } catch (CloneNotSupportedException e) { // this shouldn't happen, since we are Cloneable throw new InternalError(); } }
最后还有序列化的2个API没讲到,大家可以自己看。有些地方虽然自己理解了,但是很难写出来,但愿写清楚了。
欢迎转载,请注明出处:www.cnblogs.com/lulipro
为了获得更好的阅读体验,请访问原博客地址。
限于本人水平,如果文章和代码有表述不当之处,还请不吝赐教。
代码钢琴家
作者:lulipro-代码钢琴家
出处:http://www.cnblogs.com/lulipro/
本文版权归作者和博客园共有,欢迎转载。限于本人水平,如果文章和代码有表述不当之处,还请不吝赐教。