Java中交集、并集、差集的实现

在我们的日常开发的过程中，经常会遇到一些问题需要我们进行一些集合的运算，这里我们对集合中最为常用的交集、并集、差集这几种运算进行了一个简单的总结：

交集的实现

/**
     * 计算交集
     * @param s1
     * @param s2
     * @return
     */
    public static Set<Object> intersection(Set<Object> s1,Set<Object> s2){
        Set<Object> result = new HashSet<Object>();
        result.clear();
        result.addAll(s1);
        result.retainAll(s2);
        return result;
    }

测试及结果

public static void main(String[] args) {
        
        Set<Object> s1 = new HashSet<Object>();
        s1.add(1);
        s1.add(2);
        s1.add(3);
        Set<Object> s2 = new HashSet<Object>();
        s2.add(2);
        s2.add(3);
        s2.add(4);
        //集合中的交集测试
        Set<Object> result = TestController.intersection(s1,s2);
        if (!result.isEmpty()) {
            for (Object o : result) {
                System.out.println(o.toString());
            }
        }
    }

运行结果：

Java8 Stream 中交集的实现

public static void main(String[] args) {

        List<String> list1 = Lists.newArrayList("a", "b", "c");

        List<String> list2 = Lists.newArrayList("a", "b", "d", "e");

        //交集
        List<String> list3 = list1.stream().filter(list2::contains).collect(Collectors.toList());
        System.out.println(list3);
        

    }

并集的实现

<pre name="code" class="java">/**
     * 计算并集
     * @param s1
     * @param s2
     * @return
     */
    public static Set<Object> union(Set<Object> s1,Set<Object> s2){
        Set<Object> result = new HashSet<Object>();
        result.clear();
        result.addAll(s1);
        result.addAll(s2);
        return result;
    }

 

测试及结果

public static void main(String[] args) {

        Set<Object> s1 = new HashSet<Object>();
        s1.add(1);
        s1.add(2);
        s1.add(3);
        Set<Object> s2 = new HashSet<Object>();
        s2.add(2);
        s2.add(3);
        s2.add(4);
        //集合中的并集测试
        Set<Object> result = TestController.union(s1,s2);
        if (!result.isEmpty()) {
            for (Object o : result) {
                System.out.println(o.toString());
            }
        }
    }

运行结果：

差集的实现

/**
     * 计算差集
     * @param s1
     * @param s2
     * @return
     */
    public static Set<Object> difference(Set<Object> s1,Set<Object> s2){
        Set<Object> result = new HashSet<Object>();
        result.clear();
        result.addAll(s1);
        result.removeAll(s2);
        return result;
    }

测试及结果

public static void main(String[] args) {

        Set<Object> s1 = new HashSet<Object>();
        s1.add(1);
        s1.add(2);
        s1.add(3);
        Set<Object> s2 = new HashSet<Object>();
        s2.add(2);
        s2.add(3);
        s2.add(4);
        //集合中的并集测试
        Set<Object> result = TestController.difference(s1,s2);
        if (!result.isEmpty()) {
            for (Object o : result) {
                System.out.println(o.toString());
            }
        }
    }

运行结果

Java8 Stream 差集的实现

public static void main(String[] args) {
        List<String> list1 = Lists.newArrayList("a", "b", "c");

        List<String> list2 = Lists.newArrayList("a", "b", "d", "e");

        
        //差集: list1 - list2
        List<String> list4 = list1.stream().filter(i -> !list2.contains(i)).collect(Collectors.toList());
        System.out.println(list4);

        //差集: list2 - list1
        List<String> list5 = list2.stream().filter(i -> !list1.contains(i)).collect(Collectors.toList());
        System.out.println(list5);

      

    }