{ }codelper
集合列表排序P0099

并集、交集、差集

第一行给出 n,第二行给出 n 个整数(组 A),第三行给出 m,第四行给出 m 个整数(组 B)。
同一组内同一个数可能出现多次,但作为集合只算一次。
输出三行。
1.
A 与 B 的并集(至少在其中一个里出现的数) 2.
A 与 B 的交集(两个里都有的数) 3.
A 减 B 的差集(在 A 中而不在 B 中的数) 每行每个数只输出一次,从小到大用一个空格隔开。
若没有符合的数,输出一个 "-"。

输入
1 ≤ n, m ≤ 5,000,-2,000,000,000 ≤ 每个整数 ≤ 2,000,000,000
输出
三行(并集、交集、差集)
样例 1 · stdin
7
10 9 2 100 -1 -20 3
5
9 100 -20 7 7
stdout
-20 -1 2 3 7 9 10 100
-20 9 100
-1 2 3 10
样例 2 · stdin
3
5 5 5
1
5
stdout
5
5
-
main.py
输入(stdin)

运行时这些行会依次作为 input()/标准输入传入(运行中不会等待输入)。

$ 就绪

请在电脑上打开以编写和运行代码。在手机上可以查看题目和题解。

其他语言怎么写?查看各语言题解 · 8 种语言题解

集合运算的名字因语言而异。
Python 用运算符 a | b、a & b、a - b,Java 用 addAll、retainAll、removeAll,C# 用 UnionWith、IntersectWith、ExceptWith。
Java 和 C# 的方法会修改原集合,所以要在副本上调用。
C++ 的 std::set_union、set_intersection、set_difference 接受已排序的区间,正好配合 std::set(总是按大小排序)。
JS/TS 的 Set 很长时间没有集合运算(最近才加入 union 等),常用 filter(x => b.has(x))。
也要注意顺序。
Python 的 set 遍历顺序不是大小顺序,需要 sorted;JS/TS 的 sort() 不传比较函数时按文本排序,10 会排在 9 前面,要写 sort((x, y) => x - y)。
Java TreeSet 和 C# SortedSet 总是按大小排序。
C 没有集合,要把两组都排序、去掉重复,再从头一起扫描(归并)。
这时如果 qsort 的比较函数返回 a - b,±20 亿相减会使 int(约 ±21 亿)溢出,顺序就错了,要写 (a > b) - (a < b)。
用列表求差集([x for x in a if x not in b])会留下重复值,这也是常见错误。
Craykon 把 字典() 的键当作集合,用 包含 判断,把 键列表 排序() 后用 连接(" ") 输出。

Pythonmain.py · 14 行
input()a = set(map(int, input().split()))input()b = set(map(int, input().split()))  def show(s):    # 집합에는 차례가 없으니 출력 전에 크기 순으로 늘어놓는다    return " ".join(map(str, sorted(s))) if s else "-"  print(show(a | b))print(show(a & b))print(show(a - b))
JavaScriptmain.js · 9 行
const lines = require("fs").readFileSync(0, "utf8").split("\n");const a = new Set(lines[1].trim().split(/\s+/).map(Number));const b = new Set(lines[3].trim().split(/\s+/).map(Number));// sort()만 쓰면 글자 순(10이 9보다 앞)이라 비교 함수를 준다const show = (s) => (s.size ? [...s].sort((x, y) => x - y).join(" ") : "-");const union = new Set([...a, ...b]);const inter = new Set([...a].filter((x) => b.has(x)));const diff = new Set([...a].filter((x) => !b.has(x)));console.log([show(union), show(inter), show(diff)].join("\n"));
TypeScriptmain.ts · 10 行
import { readFileSync } from "node:fs";const lines: string[] = readFileSync(0, "utf8").split("\n");const a = new Set<number>(lines[1].trim().split(/\s+/).map(Number));const b = new Set<number>(lines[3].trim().split(/\s+/).map(Number));// sort()만 쓰면 글자 순(10이 9보다 앞)이라 비교 함수를 준다const show = (s: Set<number>): string => (s.size ? [...s].sort((x, y) => x - y).join(" ") : "-");const union = new Set<number>([...a, ...b]);const inter = new Set<number>([...a].filter((x) => b.has(x)));const diff = new Set<number>([...a].filter((x) => !b.has(x)));console.log([show(union), show(inter), show(diff)].join("\n"));
JavaMain.java · 35 行
import java.io.BufferedReader;import java.io.IOException;import java.io.InputStreamReader;import java.util.TreeSet; public class Main {    static TreeSet<Integer> readSet(BufferedReader in) throws IOException {        in.readLine();        TreeSet<Integer> s = new TreeSet<>();        for (String t : in.readLine().trim().split("\\s+")) s.add(Integer.parseInt(t));        return s;    }     static String show(TreeSet<Integer> s) {        if (s.isEmpty()) return "-";        StringBuilder sb = new StringBuilder();        for (int x : s) sb.append(sb.length() > 0 ? " " : "").append(x);        return sb.toString();    }     public static void main(String[] args) throws IOException {        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));        // TreeSet은 늘 크기 순으로 돈다. addAll·retainAll·removeAll은 원래 집합을 바꾸니 복사해서 쓴다        TreeSet<Integer> a = readSet(in), b = readSet(in);        TreeSet<Integer> union = new TreeSet<>(a);        union.addAll(b);        TreeSet<Integer> inter = new TreeSet<>(a);        inter.retainAll(b);        TreeSet<Integer> diff = new TreeSet<>(a);        diff.removeAll(b);        System.out.println(show(union));        System.out.println(show(inter));        System.out.println(show(diff));    }}
C#Main.cs · 30 行
using System;using System.Collections.Generic;using System.Linq; class Program{    static SortedSet<int> ReadSet()    {        Console.ReadLine();        return new SortedSet<int>(Console.ReadLine().Split(' ', StringSplitOptions.RemoveEmptyEntries).Select(int.Parse));    }     static string Show(SortedSet<int> s) => s.Count == 0 ? "-" : string.Join(" ", s);     static void Main()    {        // SortedSet은 늘 크기 순으로 돈다. UnionWith 등은 원래 집합을 바꾸니 복사해서 쓴다        var a = ReadSet();        var b = ReadSet();        var union = new SortedSet<int>(a);        union.UnionWith(b);        var inter = new SortedSet<int>(a);        inter.IntersectWith(b);        var diff = new SortedSet<int>(a);        diff.ExceptWith(b);        Console.WriteLine(Show(union));        Console.WriteLine(Show(inter));        Console.WriteLine(Show(diff));    }}
C++main.cpp · 34 行
#include <algorithm>#include <iostream>#include <iterator>#include <set>#include <vector> std::set<int> read_set() {    int n, x;    std::cin >> n;    std::set<int> s;    for (int i = 0; i < n; i++) {        std::cin >> x;        s.insert(x);    }    return s;} void show(const std::vector<int>& v) {    if (v.empty()) std::cout << "-";    for (std::size_t i = 0; i < v.size(); i++) std::cout << (i ? " " : "") << v[i];    std::cout << '\n';} int main() {    // std::set은 크기 순이라 set_union 등에 바로 넘길 수 있다(정렬된 범위가 필요하다)    std::set<int> a = read_set(), b = read_set();    std::vector<int> u, in, d;    std::set_union(a.begin(), a.end(), b.begin(), b.end(), std::back_inserter(u));    std::set_intersection(a.begin(), a.end(), b.begin(), b.end(), std::back_inserter(in));    std::set_difference(a.begin(), a.end(), b.begin(), b.end(), std::back_inserter(d));    show(u);    show(in);    show(d);}
Cmain.c · 50 行
#include <stdio.h>#include <stdlib.h> // a - b 를 돌려주면 ±20억끼리 빼서 int가 넘친다 — 크기만 비교한다static int cmp(const void *pa, const void *pb) {    int a = *(const int *)pa, b = *(const int *)pb;    return (a > b) - (a < b);} // 읽어서 정렬한 뒤 겹친 값을 지운다. 남은 개수를 돌려준다static int read_set(int *s) {    int n;    scanf("%d", &n);    for (int i = 0; i < n; i++) scanf("%d", &s[i]);    qsort(s, n, sizeof s[0], cmp);    int k = 0;    for (int i = 0; i < n; i++)        if (k == 0 || s[k - 1] != s[i]) s[k++] = s[i];    return k;} static void show(const int *s, int n) {    if (n == 0) printf("-");    for (int i = 0; i < n; i++) printf(i ? " %d" : "%d", s[i]);    printf("\n");} int main(void) {    static int a[5000], b[5000], u[10000], in[5000], d[5000];    int n = read_set(a), m = read_set(b);    int nu = 0, ni = 0, nd = 0, i = 0, j = 0;    // 정렬된 두 목록을 앞에서부터 함께 훑는다(병합)    while (i < n || j < m) {        if (j == m || (i < n && a[i] < b[j])) {            u[nu++] = a[i];            d[nd++] = a[i++];        } else if (i == n || b[j] < a[i]) {            u[nu++] = b[j++];        } else {            u[nu++] = a[i];            in[ni++] = a[i];            i++;            j++;        }    }    show(u, nu);    show(in, ni);    show(d, nd);    return 0;}
Craykonmain.cray · 47 行
// 사전의 열쇠를 집합처럼 쓴다(값은 아무거나)집합읽기(){    控制台.读行();    s = 字典();    遍历 (x 在 控制台.读数字())    {        s[x] = 真;    }    返回 s;} 보이기(s){    如果 (s.数量 == 0)    {        返回 "-";    }    keys = s.键列表;    keys.排序();    返回 keys.连接(" ");} a = 집합읽기();b = 집합읽기();합 = 字典();교 = 字典();차 = 字典();遍历 (x 在 a.键列表){    합[x] = 真;    如果 (b.包含(x))    {        교[x] = 真;    }    否则    {        차[x] = 真;    }}遍历 (x 在 b.键列表){    합[x] = 真;}控制台.输出(보이기(합));控制台.输出(보이기(교));控制台.输出(보이기(차));