数组的并集、交集、差集

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_35917800/article/details/82470138

public class Test {
    public static void main(String[] args) {
        Integer[] m = { 1, 2, 3, 4, 5 };
        Integer[] n = { 3, 4, 6 };

        System.out.println("----------并集------------");
        Integer[] b = getB(m, n);
        System.out.println(Arrays.toString(b));

        System.out.println("----------交集------------");
        Integer[] j = getJ(m, n);
        System.out.println(Arrays.toString(j));

        System.out.println("----------差集------------");
        Integer[] c = getC(m, n);
        System.out.println(Arrays.toString(c));

    }

    /**
     * 求并集
     */
    private static Integer[] getB(Integer[] m, Integer[] n) {
        // 将数组转换为set集合
        Set<Integer> set1 = new HashSet<Integer>(Arrays.asList(m));
        Set<Integer> set2 = new HashSet<Integer>(Arrays.asList(n));
        // 合并两个集合
        set1.addAll(set2);
        Integer[] arr = {};
        return set1.toArray(arr);
    }

    /**
     * 求交集
     */
    private static Integer[] getJ(Integer[] m, Integer[] n) {
        List<Integer> rs = new ArrayList<Integer>();
        // 将较长的数组转换为set
        Set<Integer> set = new HashSet<Integer>(Arrays.asList(m.length > n.length ? m : n));
        // 遍历较短的数组，实现最少循环
        for (Integer i : m.length > n.length ? n : m) {
            if (set.contains(i)) {
                rs.add(i);
            }
        }
        Integer[] arr = {};
        return rs.toArray(arr);
    }

    /**
     * 求差集
     */
    private static Integer[] getC(Integer[] m, Integer[] n) {
        // 将较长的数组转换为set
        Set<Integer> set = new HashSet<Integer>(Arrays.asList(m.length > n.length ? m : n));
        // 遍历较短的数组，实现最少循环
        for (Integer i : m.length > n.length ? n : m) {
            // 如果集合里有相同的就删掉，如果没有就将值添加到集合
            if (set.contains(i)) {
                set.remove(i);
            } else {
                set.add(i);
            }
        }
        Integer[] arr = {};
        return set.toArray(arr);
    }
}

结果：
----------并集------------
[1, 2, 3, 4, 5, 6]
----------交集------------
[3, 4]
----------差集------------
[1, 2, 5, 6]