ArrayDeque源码分析

引言

队列、栈是最基础的数据结构中的两种,非常简单,但也很重要。队列的规则是:先进先出(First In First Out, FIFO);栈的规则是:先进后出(First In Last Out, FILO)。而在其它方面几乎没有差别,所以java中就有一个类同时实现了它们的功能,即:ArrayDeque,直译一下应该可以叫基于数组的双端队列。

成员变量:

transient Object[] elements; // 数据载体,长度始终为2的幂
transient int head;//头指针
transient int tail;//尾指针

这里要注意的是:

  1. 这个队列不允许元素为空;
  2. 数组至少有一个位置是空的;
  3. head跟tail之间没有绝对的大小关系;

从它的设计上来看,我们可以把它看成一个循环数组,数组里存放的元素下标区间因head与tail的关系而异,具体如下:

  1. head < tail:
    [head : tail-1]
  2. head > tail:
    [head : elements.length()-1] + [0, tail-1]
  3. head == tail:
    队列为空,也就是没有存放任何元素

两个构造方法:

public ArrayDeque() {
    elements = new Object[16];
}
public ArrayDeque(int numElements) {
    allocateElements(numElements);
}

第一个很简单,就是预申请长度为16的数组;
再看第二个,numElements是指用户指定的队列长度,它调用了allocateElements(int numElements):

private void allocateElements(int numElements) {
    int initialCapacity = MIN_INITIAL_CAPACITY;
    // Find the best power of two to hold elements.
    // Tests "<=" because arrays aren't kept full.
    if (numElements >= initialCapacity) {
        initialCapacity = numElements;
        initialCapacity |= (initialCapacity >>>  1);
        initialCapacity |= (initialCapacity >>>  2);
        initialCapacity |= (initialCapacity >>>  4);
        initialCapacity |= (initialCapacity >>>  8);
        initialCapacity |= (initialCapacity >>> 16);
        initialCapacity++;

        if (initialCapacity < 0)   // Too many elements, must back off
            initialCapacity >>>= 1;// Good luck allocating 2 ^ 30 elements
    }
    elements = new Object[initialCapacity];
}

前面几句很好理解,如果numElements太小,那么队列长度就取默认初始化长度;
后面一坨代码的意思:如果numElements大于默认初始化长度,就取最小的n使得:n = 2^k(k为整数), 且n > numElements。
求n的方法有很多:从小到大枚举k、二分k等等;可能作者觉得这些方法都太low了,又熟练掌握二进制运算技巧,于是才有了这几行代码。
下面简单分析一下代码的运算过程,注:">>>"代表按位右移:
假设a=1xxxxxxxxxxxx...(base 2, x代表该位任意为0或1)
首先a |= (a >>> 1)之后,a => 11xxxxxxxx...(最高两位是1)
然后a |= (a>>> 2): a => 1111xxxxxxxxx...(最高四位是1)
再a |= (a>>> 4): a => 11111111xxxxxxx...(最高八位是1)
........
最终,a的所有低位也都变成了1,即11111111...111(全是1)
再a++ 就变成了10000000000...000(加一之后进位,比原来的二进制串多了一位,且第一位是1,其它位都是0),这个算法不仅时间效率高,而且只用到了一个变量,真可谓是短小精悍,在HashMap里也可以看到这个算法的身影。

两个添加方法:

public void addFirst(E e) {
    if (e == null)
        throw new NullPointerException();
    elements[head = (head - 1) & (elements.length - 1)] = e;
    if (head == tail)
        doubleCapacity();
}
public void addLast(E e) {
    if (e == null)
        throw new NullPointerException();
    elements[tail] = e;
    if ( (tail = (tail + 1) & (elements.length - 1)) == head)
        doubleCapacity();
}

前面代码很简洁,刚开始做元素非空检测;加到头部时,head自减并对数组长度取余,然后将元素存在head位置;加到尾部时,将元素存放在tail,再执行tail自增并对数组长度取余。最后再检查队列空间是否已满,若已满调用doubleCapacity()将容量扩增为原先的两倍:

private void doubleCapacity() {
    assert head == tail;
    int p = head;
    int n = elements.length;
    int r = n - p; // number of elements to the right of p
    int newCapacity = n << 1;
    if (newCapacity < 0)
        throw new IllegalStateException("Sorry, deque too big");
    Object[] a = new Object[newCapacity];
    System.arraycopy(elements, p, a, 0, r);
    System.arraycopy(elements, 0, a, r, p);
    elements = a;
    head = 0;
    tail = n;
}

这里主要做了这么几件事:

  1. 先做各种合法性校验;
  2. 申请新数组(长度扩大一倍);
  3. 根据head和tail推算出元素的逻辑顺序;
  4. 保持元素原来顺序拷贝到新数组里;
  5. 重置头尾指针。

这里稍微解释一下代码里的取余逻辑:
记mod = elements.length = 2^k, a为[-1,mod]之间的一个整数,那么有:

  1. a == -1:
    a & (mod-1) == mod - 1;
  2. 0 <= a < mod:
    a & (mod - 1) == a;
  3. a == mod:
    a & (mod - 1) == 0

两个取首/尾元素并删除的方法:

   public E pollFirst() {
       int h = head;
       @SuppressWarnings("unchecked")
       E result = (E) elements[h];
       // Element is null if deque empty
       if (result == null) 
           return null;
       elements[h] = null;     // Must null out slot
       head = (h + 1) & (elements.length - 1);
       return result;
   }

   public E pollLast() {
       int t = (tail - 1) & (elements.length - 1);
       @SuppressWarnings("unchecked")
       E result = (E) elements[t];
       if (result == null)
           return null;
       elements[t] = null;
       tail = t;
       return result;
   }

它们分别就是从数组中取指定下标的元素,并修改首/尾指针,最后返回。
再看两个比较低效并且不怎么用的方法:

  1. removeFirst/LastOccurrence(Object o)方法:
  public boolean removeFirstOccurrence(Object o) {
      if (o == null)
          return false;
      int mask = elements.length - 1;
      int i = head;
      Object x;
      while ( (x = elements[i]) != null) {
          if (o.equals(x)) {
              delete(i);
              return true;
          }
          i = (i + 1) & mask;
      }
      return false;
  }

  public boolean removeLastOccurrence(Object o) {
      if (o == null)
          return false;
      int mask = elements.length - 1;
      int i = (tail - 1) & mask;
      Object x;
      while ( (x = elements[i]) != null) {
          if (o.equals(x)) {
              delete(i);
              return true;
          }
          i = (i - 1) & mask;
      }
      return false;
  }

从首/尾部开始遍历,一直到找到相同的元素为止,再调用delete(int i)将其删除。delete方法里没有太多技术含量的代码,就不赘述了。
ArrayDeque里还有一些比如Iterator(比较简单),Spliterator(没用过)的代码,这边就不分析了。

总结

总体上来说,ArrayDeque的效率还是非常高的,大部分操作如从首尾插入或者读取都是O(1)的;但是队列的容量必须为2的幂这一点可能会构成很大的浪费。
比如,假设我要用的队列里元素最多的时候数量为2k,无论我们怎么努力,它会申请2(k+1)的长度,足足浪费了一半。这与ArrayList大不相同。个人认为,数组长度必须为2的幂,它所带来唯一的好处可能就是在取余的时候可以直接用位运算(&)来替代原本的取余操作(%),提高一点运算效率,不知道今后的jdk会不会对其进行优化。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,233评论 4 360
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,013评论 1 291
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,030评论 0 241
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,827评论 0 204
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,221评论 3 286
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,542评论 1 216
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,814评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,513评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,225评论 1 241
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,497评论 2 244
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,998评论 1 258
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,342评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,986评论 3 235
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,055评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,812评论 0 194
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,560评论 2 271
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,461评论 2 266

推荐阅读更多精彩内容