HashMap源码简析

说到HashMap相信大家并不陌生,这是一个非常常用的以键值对形式存储的数据结构,但是对其内部原理可能不是很了解,它的内部是以什么形式存储的,它的存取的性能号称能达到O(1)又是如何实现的,我们从源码来做个简要的分析。

主要成员变量

我们先来看以下HashMap主要有哪些成员变量

//最小值
private static final int MINIMUM_CAPACITY = 4;
//最大值,2的30次方
private static final int MAXIMUM_CAPACITY = 1 << 30;
//加载因子
static final float DEFAULT_LOAD_FACTOR = .75F;
//大小
transient int size;
//独立的entry,专用用来处理key为null的情况
transient HashMapEntry<K, V> entryForNullKey;
//加载的阀值,当size超过这个值时,将会调用doubleCapacity()方法扩容
private transient int threshold;
//存储数据的主要容器,一个HashMapEntry数组
transient HashMapEntry<K, V>[] table;
//分别是存储key和value的集合
private transient Set<K> keySet;
private transient Collection<V> values;
//存储Entry的集合
private transient Set<Entry<K, V>> entrySet;

可以看到存储数据的主要容器是一个数组,而HashMapEntry是我们循环遍历HashMap时的Entry的子类,下面我们看一下这个类

static class HashMapEntry<K, V> implements Entry<K, V> {
        final K key;
        V value;
        final int hash;
        HashMapEntry<K, V> next;

        HashMapEntry(K key, V value, int hash, HashMapEntry<K, V> next) {
            ...
        }
        ...
    }

table中的元素类型HashMapEntry,我们称之为桶,HashMapEntry里一个变量hash用于存储key的哈希值,还有一个next变量也是HashMapEntry类型,可见HashMapEntry是一个单链表结构,每一个桶都可以存储多个元素。

下面是HashMap的主要构造函数

public HashMap(int capacity) {
        ...  //参数判断和capacity为0时的空表创建

        //将capacity控制在既定的范围内并保证capacity为偶数
        if (capacity < MINIMUM_CAPACITY) {
            capacity = MINIMUM_CAPACITY;
        } else if (capacity > MAXIMUM_CAPACITY) {
            capacity = MAXIMUM_CAPACITY;
        } else {
            capacity = Collections.roundUpToPowerOfTwo(capacity);
        }
        makeTable(capacity);  //创建一个容量大小为capacity的数组
    }

makeTable方法中有如下代码:
threshold = (newCapacity >> 1) + (newCapacity >> 2);
其中threshold表示HashMap扩容的阀值,意思是在初始化时threshold等于总容量*0.75(加载因子),一旦HashMap中存储的元素数量超过threshold,就会对整个HashMap就行扩容。

扩容的方法如下

private HashMapEntry<K, V>[] doubleCapacity() {
        HashMapEntry<K, V>[] oldTable = table;
        int oldCapacity = oldTable.length;
        if (oldCapacity == MAXIMUM_CAPACITY) {
            return oldTable;  //如果旧数组的容量已经达到最大值,则直接返回
        }
        int newCapacity = oldCapacity * 2;  //将容量大小增加到以前的两倍
        HashMapEntry<K, V>[] newTable = makeTable(newCapacity);
        if (size == 0) {  //如果当前大小为空,则直接返回新数组
            return newTable;
        }

        //把oldTable中的元素重新整理到newTable中
        for (int j = 0; j < oldCapacity; j++) {
            /*
             * Rehash the bucket using the minimum number of field writes.
             * This is the most subtle and delicate code in the class.
             */
            HashMapEntry<K, V> e = oldTable[j];
            if (e == null) {
                continue;
            }
            int highBit = e.hash & oldCapacity;
            HashMapEntry<K, V> broken = null;
            newTable[j | highBit] = e;
            for (HashMapEntry<K, V> n = e.next; n != null; e = n, n = n.next) {
                int nextHighBit = n.hash & oldCapacity;
                if (nextHighBit != highBit) {
                    if (broken == null)
                        newTable[j | nextHighBit] = n;
                    else
                        broken.next = n;
                    broken = e;
                    highBit = nextHighBit;
                }
            }
            if (broken != null)
                broken.next = null;
        }
        return newTable;
    }

在重新组装table的过程使用了两个for循环遍历其中的元素,有一点复杂,我们先看HashMap的put方法,看看元素是如何存储的。

put方法

public V put(K key, V value) {
        if (key == null) {  //先处理空值情况
            return putValueForNullKey(value);
        }

        int hash = Collections.secondaryHash(key);  //计算出key的hash值
        HashMapEntry<K, V>[] tab = table;
        int index = hash & (tab.length - 1);  //计算出key的对应桶的下标
        for (HashMapEntry<K, V> e = tab[index]; e != null; e = e.next) {
            if (e.hash == hash && key.equals(e.key)) {
                preModify(e);
                V oldValue = e.value;
                e.value = value;
                return oldValue;  //key存在的情况下,直接更新value
            }
        }

        // No entry for (non-null) key is present; create one
        modCount++;
        if (size++ > threshold) {
            tab = doubleCapacity();  //size大于threshold时,调用扩容的方法,得到一个新的table
            index = hash & (tab.length - 1);
        }
        addNewEntry(key, value, hash, index);  //将新的桶更新到table中
        return null;
    }

put方法中,先计算出key的hash值,然后通过和table数组的最大index进行&操作,得到key在table里相对应的下标index,然后针对table[index]进行链表遍历,通过比较hash值和equals方法查找桶内是否存在相同的key,如果存在,则将新的value代替旧的存入,将旧的vaule返回;否则创建一个新的Entry,作为这个桶的头节点,最后返回空值。其实doubleCapacity方法只是将oldTable的遍历和newTable的定位及插入逻辑合并到了一起,原理跟put方法是十分类似的。

然后我们来看一下put方法的时间复杂度,该方法其实分为4个步骤,1.根据key计算出其hash值,并计算得到相应桶的下标index。2.查找table中index下标的Entry。3.遍历Entry为头节点的链表得到我们要找的Entry。4.取出Entry的value。
其中1、2、4步都能在O(1)时间内完成。我们之前了解到当size超过总容量*0.75后,会进行扩容,所以大部分情况下(key的hash分布较为均匀),每个桶中的元素个数 <= 1,极少数情况 >=2,所以第三步也可以在O(1)内完成,我们可以认为其整体时间复杂度为O(1)。

get方法的逻辑跟put基本一样,有兴趣可以自行了解一下。

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

推荐阅读更多精彩内容

  • 一、HashMap概述 HashMap基于哈希表的Map接口的实现。此实现提供所有可选的映射操作,并允许使用nul...
    小陈阿飞阅读 635评论 0 2
  • 实际上,HashSet 和 HashMap 之间有很多相似之处,对于 HashSet 而言,系统采用 Hash 算...
    曹振华阅读 2,510评论 1 37
  • HashMap 是 Java 面试必考的知识点,面试官从这个小知识点就可以了解我们对 Java 基础的掌握程度。网...
    野狗子嗷嗷嗷阅读 6,662评论 9 107
  • HashMap 是基于哈希表的 Map 接口的非同步实现。此实现提供所有可选的映射操作,并允许使用 null 值和...
    正规程序员阅读 494评论 0 51
  • 作为一个生来就是河蟹的我,这一生便是游走在整个世界地图上的某一角落。 偶尔会被生活所迫离开我喜欢的沼泽地,来到从未...
    艾欧尼亚的河蟹阅读 130评论 0 2