1. 程式人生 > >對ArrayList元素分組去重

對ArrayList元素分組去重

概述

通常使用SQL查詢一批資料的時候,可以利用SQL中的GROUP BY語句對資料進行分組,但是有時候出於對效能的考慮,不會使用GROUP BY,而是先把資料撈出來後,使用程式碼,在記憶體中按照某個屬性進行分組。

當然有時候還會對分組裡面的資料進行去重,這個時候也可以利用覆蓋物件的hashCodeequals方法,利用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> 裡面了。使用一個演算法按照skuIdList<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相同的已經進行去重了。