内容详情 您现在的位置是: 首页> 其他随笔

ArrayList 从源码角度剖析底层原理

发布时间:2022-02-28 19:27 已围观:2268

摘要ArrayList 从源码角度剖析底层原理


本篇文章已放到 Github github.com/sh-blog 仓库中,里面对我写的所有文章都做了分类,更加方便阅读。同时也会发布一些职位信息,持续更新中,欢迎 Star

对于 ArrayList  来说,我们平常用的最多的方法应该就是 addremove 了,本文就主要通过这两个基础的方法入手,通过源码来看看 ArrayList 的底层原理。

add

默认添加元素

这个应该是平常用的最多的方法了,其用法如下。

接下来我们就来看看 add 方法的底层源码。

ensureCapacityInternal 作用为:保证在不停的往 ArrayList 插入数据时,数组不会越界,并且实现自动扩容。

这里的 minCapacaity 的值,实际上就是在调用完当前这次 add 操作之后,数组中元素的数量

注意,这里说当前这次 add 操作完成。举个例子,调用 add 之前,ArrayList 中有3个元素,那么此时这个 minCapacity 的值就为 4

此外,可以看到将函数 calculateCapacity 的返回值作为了 ensureExplicitCapacity 的输入。

calculateCapacity 做了什么,用大白话来说是,如果当前数组是空的,则直接返回 数组默认长度(10)minCapacity 的最大值,否则就直接返回 minCapacity

接下里是 ensureExplicitCapacity ,源码如下:

modCount 表示该 ArrayList 被更改过多少次,这里的更改不只是新增,删除也是一种更改。通过上面的了解我们知道,如果当前数组内的元素个数是小于数组长度的,则 minCapacity 的值一定是小于 elementData.length 的。

相反,如果数组内的元素个数已经和数组长度相等了,则 0>0 一定是 false ,此时就会调用 grow 函数来进行数组扩容。

核心逻辑很简单,新的数组长度 = 旧数组长度 + 旧数组长度/2。

这里的右移,就相当于直接除以2

所以通过这里的源码我们验证,ArrayList 的扩容是每次扩容 1.5 倍。但是这里会有一个疑问,因为上文提到扩容时 minCapacity 的值和数组长度应该是相等的,所以 新数组长度 - minCapacity 应该永远大于0才对,为什么会有小于0的情况?

答案是 addAll

可以看到,addaddAll 底层都会调用 ensureCapacityInternaladd 是往数组中添单个元素,而 addAll 则是往数组中添加整个数组。假设 addAll 我们传进了一个很大的值,举个例子,ArrayList 的默认数组长度为 10 ,扩容一次之后长度为 15 ,假设我们传入的数组元素有 10 个,那么即使扩容一次还是不足以放下所有的元素,因为 15 < 20

所以才会出现 newCapacity(扩容之后的数组长度) < minCapacity(执行完当前操作之后的数组内元素数量)  的情况,所以当这种情况出现之后,就会直接将 minCapacity 的值赋给 newCapacity

除此之外,还会有个极端的情况,假设 addAll 往里面塞入了 Integer.MAX_VALUE 个元素呢?这就是 hugeCapacity 函数要处理的逻辑了。首先如果溢出了就直接抛出 OOM 异常,其次会保证其容量不会超过 Integer.MAX_VALUE

最后是真正执行扩容的操作,调用了 java.util 包里的 Arrays.copyOf 方法。从上图可以看出,这个方法中传入了两个参数,分别是存放元素的数组、新的数组长度,举个例子:

int[] elementData = {12345}; 
int[] newElementData = Arrays.copyOf(elementData, 10);
System.out.println(newElementData); // [1 2 3 4 5 0 0 0 0 0]

数组扩容完成之后,就会将本次 add 的元素写入 elementData 的末尾了,elementData[size++] = e

接下来我们用流程图来总结一下 add 操作的整个核心流程。

指定添加元素的位置

了解完了 addaddAll,我们趁热打铁,来看看可以指定元素位置的 add ,其接受两个参数,分别是:

  1. 新元素在数组中的下标
  2. 新元素本身

这里和最开始的  add  就有些不同了,之前的 add 方法会将元素放在数组的末尾,而 add(int index, E element) 则会将元素插入到数组中指定的位置,接下来从源码层面看看。

首先,由于这个方法允许用户传入数组下标,所以首先要做的事情就是检查传入的数组下标是否合法,如果不合法则会直接抛出 IndexOutOfBoundsException 异常。

很简单的判断,下标 index 不能小于0,并且不能超过数组中的元素个数,举个例子:

像上图这种情况,调用 add(int index, E element) 之前,数组内有3个元素,即使底层数组的长度为10 ,但是数组下标如果传入5,是会抛出 IndexOutOfBoundsException 异常的。在上图这种情况,index 的值最大只能为3才不会报错,因为 index(下标为3) > size(3个元素)  肯定不为 true

完成了校验之后,还是会调用上面聊到过的 ensureCapacityInternal 方法,根据需要,来对底层的数组进行扩容。然后调用 System.arraycopy 方法,这个方法比较关键,也比较难理解,所以我就简单一句话把它的作用概括了——将制定下标后的元素全部往后移动一位。

System.arraycopy 接收 4 个参数,分别是:

  1. 原数组
  2. 原数组中的起始下标
  3. 目标数组
  4. 目标数组中的起始下标

光看参数,不结合例子,其实很难理解,我这里举个简单的例子。

假设现在数组里有元素 [1 2 3] ,然后此时我调用方法 add(1, 4) ,表明我想要将元素 4 插入到数组下标为 1  的位置,那么此时 index 的值为1,size 的值为 3。

System.arraycopy 的方法就会变成 System.arraycopy(elementData, 1, elementData, 2, 2),也就是将 elementData  从下标 1 开始的两个元素(也就是 2 和 3),拷贝到 elementData 的从下标 2 开始的地方。

可能还是有点绕,说人话就是执行完后,elementData 就变成了 [1 2 2 3],然后再对 elementData 进行赋值,将下标为 1 的元素改为本次需要 add 的元素。再说句人话就变成了 [1 4 2 3]

所以综上来看,没有什么黑魔法,主要需要了解的就是两个关键的函数,分别是 Arrays.copySystem.arraycopy 。我们需要把这两个封装好的函数的作用给记住。

  • Arrays.copy 数组扩容
  • System.arraycopy 将数组中某个下标之后元素全部往后移动一位

所以从这里就可以看到,看源码的好处,主要有两个方面。第一,我相信你在刷题的时候一定也遇到过需要将数组的元素整个后移的 case,但是你可能并不知道可以使用 System.arraycopy ,就算你知道有这么个函数可能就连参数都看不懂;第二,知道了 System.arraycopy ,但是觉得这些函数完全没有应用场景。

remove

了解数据怎么来,接下来我们来看一下数据是怎么被移除的。首先我们来看最常用的两种:

  • 按照下标移除
  • 根据元素移除

根据下标移除

首先是根据下标移除

这里也会先检查传入的 index 是否合法,但是这里的 indexadd 中调用的 rangeCheck 还有点区别。add 中的 rangeCheckForAdd 会判断 index 是否为负数,而 remove  中的 rangeCheck 则只会判断 index 是否 >=  数组中的元素个数。

其实从函数的名称就能看出,rangeCheckForAdd 是专门给 add 方法用的

那如果此时传入的 index 真的是负数怎么办?其实是会抛出 ArrayIndexOutOfBoundsException ,因为remove 方法上加了 Range 注解。

完成后,还是会更新 modCount 的值,这也验证了我们上面提到的 modCount 代表的更改中也包含了删除。

接下来会计算一个 numMoved ,代表需要被移动的元素数量。add 一个元素,对应的下标的元素都需要往后顺移一位,remove 同理,删除了某个位置的元素后,其后面对应的所有的元素都需要往前顺移一位,就像这样:

知道了需要移动多少个元素之后,我们的 System.arraycopy 就又可以登场了。完成了元素的移动之后,数组的末尾必然会空出来一个元素,直接将其设置为 null 然后交给 GC 回收即可,最后把被移除的值返回。

根据值移除

举个例子,根据值移除就长下面这样这样。

废话不多说,直接看核心源码

完了,第二行就给整懵了,移除一个 null 是什么鬼?还要循


声明:本文内容摘自网络,版权归原作者所有。如有侵权,请联系处理,谢谢~
转发:detectiveHLH--https://www.cnblogs.com/detectiveHLH/p/15033621.html

赞一个 (354)