對ArrayList元素分組去重
阿新 • • 發佈:2019-02-03
概述
通常使用SQL
查詢一批資料的時候,可以利用SQL
中的GROUP BY
語句對資料進行分組,但是有時候出於對效能的考慮,不會使用GROUP BY
,而是先把資料撈出來後,使用程式碼,在記憶體中按照某個屬性進行分組。
當然有時候還會對分組裡面的資料進行去重,這個時候也可以利用覆蓋物件的hashCode
和equals
方法,利用HashSet
的去重特性來達到目的
程式碼
public class SkuVo {
private Long skuId;
private String productName;
private Long brandStoreSn;
public SkuVo(Long skuId, String productName, Long brandStoreSn) {
super();
this.skuId = skuId;
this.productName = productName;
this.brandStoreSn = brandStoreSn;
}
public Long getSkuId() {
return skuId;
}
public void setSkuId(Long skuId) {
this .skuId = skuId;
}
public String getProductName() {
return productName;
}
public void setProductName(String productName) {
this.productName = productName;
}
public Long getBrandStoreSn() {
return brandStoreSn;
}
public void setBrandStoreSn(Long brandStoreSn) {
this .brandStoreSn = brandStoreSn;
}
@Override
public String toString() {
return "SkuVo [skuId=" + skuId + ", productName=" + productName + ", brandStoreSn=" + brandStoreSn + "]";
}
@Override
public int hashCode() {
final int prime = 31;
int result = 1;
result = prime * result + ((brandStoreSn == null) ? 0 : brandStoreSn.hashCode());
return result;
}
@Override
public boolean equals(Object obj) {
if (this == obj)
return true;
if (obj == null)
return false;
if (getClass() != obj.getClass())
return false;
SkuVo other = (SkuVo) obj;
if (brandStoreSn == null) {
if (other.brandStoreSn != null)
return false;
} else if (!brandStoreSn.equals(other.brandStoreSn))
return false;
return true;
}
}
假設從資料查詢出一批資料,存在了List<SkuVo>
裡面了。使用一個演算法按照skuId
對List<SkuVo>
進行分組,skuId
相同的歸為一組。如果小組裡面的資料,brandStoreSn
出現重複,則進行去重操作。
分組去重演算法
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
public class TestArrayListGroupByTwoKey {
public static void main(String[] args) {
/*1、準備資料**/
SkuVo sku1 = new SkuVo(1L,"p1",100L);
SkuVo sku2 = new SkuVo(2L,"p2",101L);
SkuVo sku5 = new SkuVo(2L,"p5",100L);
SkuVo sku3 = new SkuVo(3L,"p3",102L);
SkuVo sku4 = new SkuVo(3L,"p3",102L);
SkuVo sku6 = new SkuVo(5L,"p6",100L);
List<SkuVo> skuVoList = Arrays.asList(new SkuVo [] {sku1,sku2,sku3,sku4,sku5,sku6});
/*2、分組演算法**/
Map<Long, Set<SkuVo>> skuIdMap = new HashMap<>();
for (SkuVo skuVo : skuVoList) {
Set<SkuVo> tempList = skuIdMap.get(skuVo.getSkuId());
/*如果取不到資料,那麼直接new一個空的HashSet**/
if (tempList == null) {
tempList = new HashSet<>();
tempList.add(skuVo);
skuIdMap.put(skuVo.getSkuId(), tempList);
}
else {
/*某個sku之前已經存放過了,則直接追加資料到原來的Set裡,
*如果brandStoreSn重複,利用HashSet的特性進行去重**/
tempList.add(skuVo);
}
}
/*3、遍歷map,驗證結果**/
for(Long skuId : skuIdMap.keySet()){
System.out.println(skuIdMap.get(skuId));
}
}
}
結果如下
[SkuVo [skuId=1, productName=p1, brandStoreSn=100]]
[SkuVo [skuId=2, productName=p5, brandStoreSn=100], SkuVo [skuId=2, productName=p2, brandStoreSn=101]]
[SkuVo [skuId=3, productName=p3, brandStoreSn=102]]
[SkuVo [skuId=5, productName=p6, brandStoreSn=100]]
從輸出結果看,資料已經按照skuId
進行分組了,同時小組內brandStoreSn相同的已經進行去重了。