如何用java8中stream去重？

Question

我有一个List<List<Integer>> res,比如res的值为

[
[2, 2, 2, 2]
[2, 3, 3]
[3, 2, 3]
[3, 3, 2]
[3, 5]
[5, 3]
]

我如何使用stream流去除res里面的和一致的元素？
比如[2, 3, 3]、[3, 2, 3]、[3, 3, 2]这三个的和都为8，只需保留一个。

Answer 1

有两个方法实现

因为好久没写 Java，所以语法和库都不是很熟，尽请指出代码优化方案

方法一

使用 Stream<T>.distinct() 来取唯一值，不过它是根据 T 的 equals() 和 hashCode() 来判断重复的，所以需要写一个类来包装 List<Integer> 提供按 sum 判断唯一的方法

class Wrapper {
    private List<Integer> data;
    private int sum;

    public Wrapper(List<Integer> data) {
        this.data = data;
        sum = data.stream().collect(Collectors.summingInt(Integer::intValue));
    }

    public List<Integer> getData() {
        return data;
    }

    @Override
    public int hashCode() {
        return sum;
    }

    @Override
    public boolean equals(Object obj) {
        if (obj instanceof Wrapper) {
            return equals((Wrapper) obj);
        }

        return false;
    }

    public boolean equals(Wrapper wrapper) {
        return sum == wrapper.sum;
    }
}

然后：先封装 ⇒ 计算唯一 ⇒ 解封装 ⇒ 生成 List

private static List<List<Integer>> distinct1(List<List<Integer>> data) {
    return data.stream()
            .map(intValues -> new Wrapper(intValues))
            .distinct()
            .map(w -> w.getData())
            .collect(Collectors.toList());
}

方法二

直接通过 Stream<T>.filter() 来过滤掉重复的，需要一个 HashSet 来保存应该判重的值。这个方法不需要另外定义一个类，只需要一个临时对象。

private static List<List<Integer>> distinct2(List<List<Integer>> data) {
    HashSet<Integer> cache = new HashSet<>();
    return data.stream()
        .filter(intValues -> {
            Integer sum = intValues.stream().collect(Collectors.summingInt(Integer::intValue));
            if (cache.contains(sum)) {
                return false;
            }

            cache.add(sum);
            return true;
        })
        .collect(Collectors.toList());
}

---

已参与了 SegmentFault 思否「问答」打卡，欢迎正在阅读的你也加入。