HashMap 的 get 实现?
相对于 put 来说,get 比较简单:
- 计算 key 的 hash 值,根据 hash 值找到对应数组下标: hash & (length-1)
- 判断数组该位置处的元素是否刚好就是我们要找的,如果不是,走第三步
- 判断该元素类型是否是 TreeNode,如果是,用红黑树的方法取数据,如果不是,走第四步
- 遍历链表,直到找到相等(==或equals)的 key
public V get(Object key) {
Node<K,V> e;
return (e = getNode(hash(key), key)) == null ? null : e.value;
}
final Node<K,V> getNode(int hash, Object key) {
Node<K,V>[] tab; // 哈希表数组
Node<K,V> first, e; // first 表示桶中的第一个节点,e 用于遍历链表或树中的节点
int n; // 哈希表容量
K k; // 键对象
// 检查哈希表非空,容量大于0,以及哈希值所在桶不为空
if ((tab = table) != null && (n = tab.length) > 0 &&
(first = tab[(n - 1) & hash]) != null) {
// 检查第一个节点是否匹配,一般在桶的第一个节点开始查找
if (first.hash == hash && // 始终检查第一个节点
((k = first.key) == key || (key != null && key.equals(k))))
return first;
// 如果有多个节点在同一个桶中,需要遍历链表或树
if ((e = first.next) != 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);
}
}
// 如果未找到匹配的键值对,返回 null
return null;
}