关于set集合不重复原理

陈救驾来迟~

于 2022-09-25 22:38:09 发布

阅读量481

点赞数 1

文章标签：哈希算法散列表算法

本文链接：https://blog.csdn.net/cqq19990909/article/details/127044435

版权

1.底层原理

首先，我们要先了解一下set集合的底层结构，底层结构是由数组+链表+红黑树。

2.存入元素时的不重复原理

set集合存入元素时，首先会根据hashcode方法计算出该元素的hash值，如果已经存在该hash值，则会调用eaqul方法，判断两个元素是否相同，相同则不存；如果hash值相同，元素不相同，则会形成链表直接把元素挂在已经存在的元素下边；如果hash值不相同则会直接存进去。下来我们看看源码：

public boolean add(E e) {
return map.put(e, PRESENT)==null;
}
查看源码我们可以发现，HashSet其实是调用了map集合中的put方法来保证元素不会重复。

再点进去看看put源码：

public V put(K key, V value) {
return putVal(hash(key), key, value, false, true);
}
会发现这里面又调用了putVal方法和hash方法，那么现在把这些方法放在一起来看看：

public class HashMap<K,V> implements Map<K,V>{
   //hash()方法是计算元素的哈希值，它底层是hashCode()方法
       //如果传进来的元素是null，返回哈希值为0
       //如果不是null，则按后面的表达式计算哈希值
static final int hash(Object key) {
int h;
return (key == null) ? 0 : (h = key.hashCode()) ^ (h >>> 16);
}

//Set集合实现去重，底层调用的是HashMap中的put()方法，而put方法调用的是HashMap中的putVal()方法
public V put(K key, V value) {//"hello"
return putVal(hash(key), key, value, false, true);
}

//putVal方法
final V putVal(int hash, K key, V value, boolean onlyIfAbsent,boolean evict) {
//哈希表存储的是一个个的结点数组
Node<K,V>[] tab; Node<K,V> p; int n, i;

//判断哈希表是否初始化，如果没有初始化，我们就对其进行初始化
if ((tab = table) == null || (n = tab.length) == 0)
n = (tab = resize()).length;

/*
根据元素之前计算得出的哈希值，计算该元素在哈希表中的存储位置
如果该位置是null，表示为空，可以创建新的节点，存储元素
*/
if ((p = tab[i = (n - 1) & hash]) == null)
tab[i] = newNode(hash, key, value, null);

else {
Node<K,V> e; K k;

       /*
           如果元素存储位置不为空，也就是已经有元素了，则会进行比较
           将要存入的元素称为新元素，将该位置上原先就有的元素称为旧元素
           如果哈希值不同，说明新元素与旧元素不同，可以将新元素加入到集合中
           如果哈希值相同，则继续调用对象的equals方法进行比较
               如果equals方法比较的结果为false，说明两者不同，将新元素加入到集合中
               如果equals方法比较的结果为true，说明两者完全相同，不会添加新元素
       */
if (p.hash == hash &&
((k = p.key) == key || (key != null && key.equals(k))))
e = p;
else if (p instanceof TreeNode)
e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
else {
for (int binCount = 0; ; ++binCount) {
if ((e = p.next) == null) {
p.next = newNode(hash, key, value, null);
if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
treeifyBin(tab, hash);
break;
}
if (e.hash == hash &&
((k = e.key) == key || (key != null && key.equals(k))))
break;
p = e;
}
}
if (e != null) { // existing mapping for key
V oldValue = e.value;
if (!onlyIfAbsent || oldValue == null)
e.value = value;
afterNodeAccess(e);
return oldValue;
}
}
++modCount;
if (++size > threshold)
resize();
afterNodeInsertion(evict);
return null;
}
}