对ArrayList元素分组去重

最新推荐文章于 2024-12-16 10:31:45 发布

原创最新推荐文章于 2024-12-16 10:31:45 发布 · 1.9k 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#arraylist

本文介绍了一种不使用SQL GROUP BY语句而在Java内存中实现数据分组与去重的方法，通过自定义对象的hashCode和equals方法，并利用HashSet的特性完成去重。

概述

通常使用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相同的已经进行去重了。