Java Set集合去重

Java Set集合去重

在开发中经常使用到Set集合去重,那么去重的原理是怎样实现的呢?在此文章记录一下去重原理!!!

下面是set集合类图
image.png
下面我们来跟踪一下执行过程;

  1. 首先我们实例化一个Set对象;
    Set<8大基本类型> set = new HashSet<8大基本类型>();
    set.add(8大基本类型);

  1. add操作会调用HashMap中的add方法;
public boolean add(E e) {
	return map.put(e, PRESENT)==null;
	}

  1. HashMap中的add方法依赖了HashMap的put方法;
public V put(K key, V value) {
        if (key == null)
            return putForNullKey(value);
        int hash = hash(key.hashCode());
        int i = indexFor(hash, table.length);
        for (Entry<K,V> e = table[i]; e != null; e = e.next) {//每添加一个,则循环判断是否与map中的元素相等
            Object k;
            // 先判断hashcode是否一致,然后再判断值是否相等
            if (e.hash == hash && ((k = e.key) == key || key.equals(k))) {
                V oldValue = e.value;
                e.value = value;
                e.recordAccess(this);
                return oldValue;
            }
        }
 
        modCount++;
        addEntry(hash, key, value, i);
        return null;
    }

但是上述方法只对基本数据类型有效,下面是应用到复杂对象的讲解。
下面就是重写对象User的实现,重写equals和hashCode方法;

测试类

public class User {
    //id
    protected Integer id;
    //username
    protected String username;

    //构造方法
    public User(int id,String username){
        this.id = id;
        this.username = username;
    }

    /**
     * 如果对象是USER,先比较hashcode,一致的场合在比价每个属性的值
     */
    @Override
    public boolean equals(Object obj) {
        if(obj == null)
            return false;
        if(this == obj)
            return true;
        if(obj instanceof User){
            User user = (User) obj;
            //if(user.id == this.id) return true; //只比较id
            //比较id和username 一致时才返回true,之后再去比较hashCode
            if(user.id == this.id && user.username.equals(this.username)){
                return true;
            }
        }
        return false;
    }

    /**
     *
     * 重写hashCode方法,返回的hashCode不一样 才认为是不一样的对象;
     */
    @Override
    public int hashCode() {
        //return id.hashCode();  只比较id,id不一样就添加进集合;
        return id.hashCode() * username.hashCode();
    }
}

实现类

import java.util.HashSet;
import java.util.Set;

public class test {
    public static void main(String[] args){
        User user1 = new User(1,"xiaoqiang");
        User user2 = new User(2,"xiaoqiang");
        User user3 = new User(1,"xiaoqiang");
        User user4 = new User(1,"xiaoqiang");

        Set<User> set = new HashSet<User>();
        set.add(user1);
        set.add(user2);
        set.add(user3);
        set.add(user4);
        for(User u : set){
            System.out.println("id:" + u.id +" username:"+ u.username);
        }
    }
}

输出结果

id:2 username:xiaoqiang
id:1 username:xiaoqiang