Java笔记 —— Set集合的去重原理(源码分析)

public boolean add(E e) {

return map.put(e, PRESENT)==null;

}

再点进去查看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;

}

}

因此简单来说，Set集合实现去重的原理是

先看元素的HashCode()值也就是哈希值是否相同，再看equals方法返回的结果

如果哈希值不同，则说明元素不相同，将元素添加到集合中

如果哈希值相同，则继续判断equals方法

————如果equals返回true，则说明元素的哈希值与内容都相同，元素重复，不会添加到集合中

————如果equals返回false，则说明元素的哈希值相同，但内容不同，元素不重复，添加到集合中

自定义类重写equals与hashcode方法

那么现在回过头来看上面的Set集合例子，传入自定义类的对象时，因为类里面没有重写equals方法和hashcode方法，导致Set集合无法正确判断两个元素是否相同，因此没有去重成功

那为什么传入的元素是String类型时，可以去重呢？

因为不同的字符串的哈希值不相同，而对于相同哈希值的字符串，String类中已经重写了equals方法，因此可以正确判断两个String类型元素是否相同。

那么现在，就来重写类里面的equals方法和hashcode方法

package review.SetDemo;

import java.util.Objects;

public class Student{

private String name;

private int age;

public Student() {

}

public Student(String name, int age) {

this.name = name;

this.age = age;

}

public String getName() {

return name;

}

public void setName(String name) {

this.name = name;

}

public int getAge() {

return age;

}

public void setAge(int age) {

this.age = age;

}

@Override

public String toString() {

return “Student{” +

“name='” + name + ‘’’ +

“, age=” + age +

‘}’;

}

@Override

public boolean equals(Object o) {

if (this == o) return true;

if (o == null || getClass() != o.getClass()) return false;

Student student = (Student) o;

return age == student.age &&

Objects.equals(name, student.name);

}

@Override

public int hashCode() {

return Objects.hash(name, age);

}

}

再回过头来运行

public class demo2 {

public static void main(String[] args) {

HashSet set = new HashSet<>();

Student s1 = new Student(“zhang”,12);

Student s2 = new Student(“chen”,20);

Student s3 = new Student(“cao”,60);

Student s4 = new Student(“zhang”,12);

set.add(s1);

set.add(s2);

set.add(s3);

最后希望可以帮助到大家！

千千万万要记得：多刷题！！多刷题！！

之前算法是我的硬伤，后面硬啃了好长一段时间才补回来，算法才是程序员的灵魂！！！！

篇幅有限，以下只能截图分享部分的资源！！

（1）多线程（这里以多线程为代表，其实整理了一本JAVA核心架构笔记集）

（2）刷的算法题（还有左神的算法笔记）

（3）面经+真题解析+对应的相关笔记（很全面）

（4）视频学习（部分）

ps：当你觉得学不进或者累了的时候，视频是个不错的选择

在这里，最后只一句话：祝大家offer拿到手软！！

最后希望可以帮助到大家！

千千万万要记得：多刷题！！多刷题！！

之前算法是我的硬伤，后面硬啃了好长一段时间才补回来，算法才是程序员的灵魂！！！！

篇幅有限，以下只能截图分享部分的资源！！

（1）多线程（这里以多线程为代表，其实整理了一本JAVA核心架构笔记集）

[外链图片转存中…(img-GFFhp8Aw-1719273701314)]

（2）刷的算法题（还有左神的算法笔记）

[外链图片转存中…(img-ZfHCJsbS-1719273701315)]

（3）面经+真题解析+对应的相关笔记（很全面）

[外链图片转存中…(img-J1TFIPo0-1719273701315)]

（4）视频学习（部分）

ps：当你觉得学不进或者累了的时候，视频是个不错的选择

在这里，最后只一句话：祝大家offer拿到手软！！