面试必问的HashMap,一次彻底帮你搞定HashMap源码,以后不用再头疼了

static final class TreeNode<K,V> extends LinkedHashMap.Entry<K,V> {
TreeNode<K,V> parent; // red-black tree links
TreeNode<K,V> left;
TreeNode<K,V> right;
TreeNode<K,V> prev; // needed to unlink next upon deletion
boolean red;
}

他的继承结构是这样,可以看到继承了Node节点

看懂一条语句

hash & tab.length-1

代码中多处都可以看到这条代码,实际上这条语句只是做了一个取余(%)的动作。一个&怎么做的取余的操作: HashMap的容量为2^n其二进制结构如下

任何数&2^n-1(01111…)其结果都是去0xxxx,做了快速取余的操作。后续会看到该条语句频繁出现

几个核心参数

  • Node<K,V>[] table:存放Node的数组
  • int size:表示当前HashMap包含的键值对数量
  • int threshold:表示当前HashMap能够承受的最多的键值对数量,一旦超过这个数量HashMap就会进行扩容
  • final float loadFactor:负载因子,用于扩容
  • int DEFAULT_INITIAL_CAPACITY = 16:默认的table初始容量
  • float DEFAULT_LOAD_FACTOR = 0.75f:默认的负载因子
  • int TREEIFY_THRESHOLD = 8: 链表长度大于该参数转红黑树
  • int UNTREEIFY_THRESHOLD = 6: 当树的节点数小于等于该参数转成链表

初始化方法

指定了具体的容量,以及负载因子的初始化方法。当知道需要放入的元素的个数时可以先指定避免多次扩容造成性能浪费。

public HashMap(int initialCapacity) {
this(initialCapacity, DEFAULT_LOAD_FACTOR);
}
public HashMap(int initialCapacity, float loadFactor) {
this.loadFactor = loadFactor;
this.threshold = tableSizeFor(initialCapacity);
}

核心方法 public V get(Object key)

参数key,以及该key的hash 先判断数组是否已经初始化了,以及数组长度。 在判断tab[(n - 1) & hash],前文提到的那一条语句,用key的hash取余数组长度判断数组中的位置是否存在元素。

  • 不存在元素 数组中不存在元素,肯定没有产生hash冲突,那么元素肯定不存在
  • 数组当前位置存在元素 判断key的hsah相等并且(key的地址相等或者equals相等)。那么可以确定元素在数组中。
  • 数组当前位置存在元素,但是key不相等 接下来会去判断数组中当前位置是否存在next元素(Node节点结构),如果有next说明存在链表或者树形结构。 接下来判断Node是否是TreeNode,如果是则按照遍历树方式遍历得到结果,不是则按照遍历链表的形式遍历得到结果。

final Node<K,V> getNode(int hash, Object key) {
Node<K,V>[] tab = table;
Node<K,V> first = tab[(n - 1) & hash];
Node<K,V> e = first.next;
int n = tab.length;
K k;
//数组是否已经初始化
if (tab!= null && n > 0 && first != null) {
//table中是否有节点,key是否相等
if (first.hash == hash && ((k = first.key) == key || (key != null && key.equals(k))))
return first;
//key不相等,判断是否有next,并且判断是树的节点还是链表的节点,再以不同的方式去遍历获取
if (e != null) {
if (first instanceof TreeNode)
return ((TreeNode<K,V>)first).getTreeNode(hash, key);
do {
if (e.hash == hash && ((k = e.key) == key || (key != null && key.equals(k))))
return e;
} while ((e = e.next) != null);
}
}
return null;
}

核心方法 public V put(K key, V value)

put方法比较长,分几种情况解析

  • 第一次put元素,数组还未初始化:调用resize()初始化数组,直接放入table相应位置。(resize扩容方法很重要)
  • table中该位置没产生Hash冲突:构造节点放入table中
  • 产生Hash冲突,先判断table中节点元素key是否相等,相等则替换value
  • 产生Hash冲突,节点是TreeNode:按红黑树的方式插入节点
  • 产生Hash冲突,节点是Node:构造节点按链表的方式插入,并且检查插入后是否到达转红黑树的阈值8

final V putVal(int hash, K key, V value, boolean onlyIfAbsent, boolean evict) {
Node<K,V>[] tab;
Node<K,V> p;
int n;
int i;
//第一次put元素,数组还未初始化
if ((tab = table) == null || (n = tab.length) == 0)
//调用resize()初始化数组
n = (tab = resize()).length;
//table中该位置没产生Hash冲突
if ((p = tab[i = (n - 1) & hash]) == null)
//构造节点放入table中
tab[i] = newNode(hash, key, value, null);
else {
//产生Hash冲突
Node<K,V> e; K k;
//table中节点元素key是否相等
if (p.hash == hash && ((k = p.key) == key || (key != null && key.equals(k))))
//记录下节点的引用,后续替换E节点的值
e = p;
else if (p instanceof TreeNode)
//节点是TreeNode:按红黑树的方式插入节点
e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
else {
//节点是Node,链表情况
for (int binCount = 0; ; ++binCount) {
//遍历到链表尾部还未发现相同的key,则构造节点插入到链表
if ((e = p.next) == null) {
p.next = newNode(hash, key, value, null);
if (binCount >= TREEIFY_THRESHOLD - 1)
//检查插入后是否到达转红黑树的阈值8
treeifyBin(tab, hash);
break;
}
//检查是否有相同的key,有就退出,e = p.next已经记录了E的引用
if (e.hash == hash &&
((k = e.key) == key || (key != null && key.equals(k))))
break;
p = e;
}
}
//上述各种情况下,如果不是插入节点的情况下
//存在key相同的情况下,完成一个值的替换
if (e != null) {
V oldValue = e.value;
if (!onlyIfAbsent || oldValue == null)
e.value = value;
return oldValue;
}
}
//检查是否需要扩容 存在的数量>hashmap容量*负载因子就需要扩容
if (++size > threshold)
resize();
return null;
}

核心方法Node<K,V>[] resize()

最重要的部分来了,也是面试官最喜欢问的HashMap的扩容

  1. 第一次初始化时候调用
  2. 存放的键值对的数量>hashmap容量*负载因子就需要扩容

初始化时候调用resize

扩容后得到的是一个Node数组。由于第一次初始化,肯定是不存在链表,红黑树等结构的,以及Node节点的。只是对一些属性做了赋值操作,和返回一个空的Node数组。 threshold:HashMap能够承受的最多的键值对数量;如果指定了容量和负载因子,则threshold = 指定的容量*负载因子; Node<K,V>[] table:存放Node的数组,创建了一个容量为16(未指定具体容量时,默认为16)的Node数组。

省略部分与第一次初始化无关代码不重要的代码

final HashMap.Node<K,V>[] resize() {
//16*0.75=12
threshold = (int)(DEFAULT_LOAD_FACTOR * DEFAULT_INITIAL_CAPACITY);
//16
HashMap.Node<K,V>[] newTab = (HashMap.Node<K,V>[])new HashMap.Node[DEFAULT_INITIAL_CAPACITY];
table = newTab;
return newTab;
}

扩容调用resize 重点来了

先删除一些与扩容无关的代码

final Node<K,V>[] resize() {
Node<K,V>[] oldTab = table;
int oldCap = (oldTab == null) ? 0 : oldTab.length;
int oldThr = threshold;
int newCap, newThr = 0;
if (oldCap > 0) {
//容量和阈值都扩大成两倍
newCap = oldCap << 1;
newThr = oldThr << 1;
}
//设置阈值属性
threshold = newThr;
//新建一个是之前两倍容量的大小的Node数组
Node<K,V>[] newTab = (Node<K,V>[])new Node[newCap];
//属性赋值
table = newTab;
//完成一些准备,开始准备迁移之前节点
if (oldTab != null) {
//循环迁移每个节点数据
for (int j = 0; j < oldCap; ++j) {
Node<K,V> e;
//并不数组中是每个位置都有元素
if ((e = oldTab[j]) != null) {
//数据需要迁移,table相应位置置空
oldTab[j] = null;
//不存在链表情况
if (e.next == null)
//重新对新的容量快速取余,放入相应的位置
newTab[e.hash & (newCap - 1)] = e;
//树结构(较为复杂后续再分析)
else if (e instanceof TreeNode)
((TreeNode<K,V>)e).split(this, newTab, j, oldCap);
//链表结构
else {
//不需要移动的链表的头尾指针
Node<K,V> loHead = null, loTail = null;
//需要移动的链表的头尾指针
Node<K,V> hiHead = null, hiTail = null;
Node<K,V> next;
//遍历链表将一个链表拆成两个链表,这里主要分析一下拆分依据
do {
next = e.next;
//扩容后余数与之前一致,不需要移动的节点。
if ((e.hash & oldCap) == 0) {
if (loTail == null)
loHead = e;
else
loTail.next = e;
loTail = e;
} else {
//扩容后余数与之前不一致,需要移动的节点。
if (hiTail == null)
hiHead = e;
else
hiTail.next = e;
hiTail = e;
}
} while ((e = next) != null);
//将链表重新放入table数组中
if (loTail != null) {
loTail.next = null;
newTab[j] = loHead;
}
if (hiTail != null) {
hiTail.next = null;
newTab[j + oldCap] = hiHead;
}
}
}
}
}
l)
hiHead = e;
else
hiTail.next = e;
hiTail = e;
}
} while ((e = next) != null);
//将链表重新放入table数组中
if (loTail != null) {
loTail.next = null;
newTab[j] = loHead;
}
if (hiTail != null) {
hiTail.next = null;
newTab[j + oldCap] = hiHead;
}
}
}
}
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值