【集合】List取交集、并集（去重）、差集 - Java技术债务

Java技术债务

发布于 2024-06-21 17:01:48

910

发布于 2024-06-21 17:01:48

交集

最笨的方法之一：双层for循环（对象的情况下可以转成map或者set）

public static void main(String[] args) {
        List listA = new ArrayList<>();
        List listB = new ArrayList<>();
        listA.add("A");
        listA.add("B");
        listA.add("C");
        listA.add("D");
        listB.add("B");
        listB.add("C");
        listB.add("D");
        listB.add("E");
        Set set = new HashSet();
        for (String tmp: listA) {
            set.add(tmp);
        }
        for (String tmp: listB) {
            if(set.contains(tmp)){
                System.out.println(tmp);
            }
        }
    }

上边代码可以简化为一行代码，使用List源码中retainAll(List list)

listA.retainAll(listB)
结果：
B,C,D

ArrayList对此方法进行了重写，先对传入的参数进行了非空判断

然后调用了batchRemove方法执行取交集逻辑

 final Object[] elementData = this.elementData;

这里要注意的是这样赋值，elementData引用还是指向this.elementData，所以后面直接操作elementData改变的会是this。

对自身进行了遍历，在遍历的过程中判断传进来的集合中是否包含当前元素，如果包含，则按顺序写入当前集合中。

第二个判断就是将交集后面多余的元素去除。

差集

public static void main(String[] args) {
        List listA = new ArrayList<>();
        List listB = new ArrayList<>();
        listA.add("A");
        listA.add("B");
        listA.add("C");
        listA.add("D");
        listB.add("B");
        listB.add("C");
        listB.add("D");
        listB.add("E");
        List listC = new ArrayList<>();
        listC.addAll(listA);
        // 方法一: for循环
        for (String s : listB) {
            listA.remove(s);
        }
        System.err.println(listA);
        // 方法二: list自带方法
        listC.removeAll(listB);
        System.err.println(listC);
    }
结果：
[A]
[A]

并集

不去重

public static void main(String[] args) {
        List listA = new ArrayList<>();
        List listB = new ArrayList<>();
        listA.add("A");
        listA.add("B");
        listA.add("C");
        listA.add("D");
        listB.add("B");
        listB.add("C");
        listB.add("D");
        listB.add("E");
				listA.addAll(listB);
    }
结果：
[A, B, C, D, B, C, D, E]

去重

public static void main(String[] args) {
            List listA = new ArrayList<>();
            List listB = new ArrayList<>();
            listA.add("A");
            listA.add("B");
            listA.add("C");
            listA.add("D");
            listB.add("B");
            listB.add("C");
            listB.add("D");
            listB.add("E");
            List listC = new ArrayList<>();
            listC.addAll(listA);
            // 方法一: for循环
            for (String s : listB) {
                if (!listA.contains(s)) {
                    listA.add(s);
                }
            }
            System.err.println(listA);
            // 方法二: list自带方法
            listC.removeAll(listB);
            listC.addAll(listB);
            System.err.println(listC);
    }
结果：
[A, B, C, D, E]
[A, B, C, D, E]

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2023-11-06，如有侵权请联系 cloudcommunity@tencent.com 删除

集合