Java Set集合去重的原理及實現

更新時間：2020年09月08日 11:29:47 作者：崔笑顏

這篇文章主要介紹了Java Set集合去重的原理及實現，幫助大家更好的理解和學習Java，感興趣的朋友可以了解下

在開發(fā)中經常使用到Set集合去重，那么去重的原理是怎樣實現的呢？在此文章記錄一下去重原理?。?！

下面是set集合類圖

下面我們來跟蹤一下執(zhí)行過程；

首先我們實例化一個Set對象;

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

add操作會調用HashMap中的add方法;

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

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) {//每添加一個，則循環(huán)判斷是否與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);
    }
  }
}

輸出結果