欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Java Set集合去重

程序员文章站 2022-04-15 14:42:18
...

Java Set集合去重

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

下面是set集合类图
Java Set集合去重
下面我们来跟踪一下执行过程;
1、首先我们实例化一个Set对象;

    Set<8大基本类型> set = new HashSet<8大基本类型>();
    set.add(8大基本类型);

2、add操作会调用HashMap中的add方法;

public boolean add(E e) {
	return map.put(e, PRESENT)==null;
	}

3、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

相关标签: 笔记 java