[백준] 1269번 - 대칭 차집합 Python, Java

문제

---

자연수를 원소로 갖는 공집합이 아닌 두 집합 A와 B가 있다. 이때, 두 집합의 대칭 차집합의 원소의 개수를 출력하는 프로그램을 작성하시오. 두 집합 A와 B가 있을 때, (A-B)와 (B-A)의 합집합을 A와 B의 대칭 차집합이라고 한다.

예를 들어, A = { 1, 2, 4 } 이고, B = { 2, 3, 4, 5, 6 } 라고 할 때, A-B = { 1 } 이고, B-A = { 3, 5, 6 } 이므로, 대칭 차집합의 원소의 개수는 1 + 3 = 4개이다.

입력

---

첫째 줄에 집합 A의 원소의 개수와 집합 B의 원소의 개수가 빈 칸을 사이에 두고 주어진다. 둘째 줄에는 집합 A의 모든 원소가, 셋째 줄에는 집합 B의 모든 원소가 빈 칸을 사이에 두고 각각 주어진다. 각 집합의 원소의 개수는 200,000을 넘지 않으며, 모든 원소의 값은 100,000,000을 넘지 않는다.

출력

---

첫째 줄에 대칭 차집합의 원소의 개수를 출력한다.

예제 입력 1

---

3 5
1 2 4
2 3 4 5 6

예제 출력 1

---

4

풀이

---

Python

import sys
input = sys.stdin.readline

A,B = map(int,input().rstrip().split())

a = set(list(map(int,input().rstrip().split())))
b = set(list(map(int,input().rstrip().split())))

dif = len(a) - len(a-b)

print(len(a)+len(b)-2*dif)

Java

import java.io.*;
import java.util.*;


public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        HashSet<Integer> set = new HashSet<>();
        int a = Integer.parseInt(st.nextToken());
        int b = Integer.parseInt(st.nextToken());
        int cnt = 0;

        st = new StringTokenizer(br.readLine());
        for(int i = 0; i < a; i++) {
            set.add(Integer.parseInt(st.nextToken()));
        }

        st = new StringTokenizer(br.readLine());
        for(int i = 0; i < b; i++) {
            int tmp = Integer.parseInt(st.nextToken());
            if (set.contains(tmp)) {
                cnt++;
            } else {
                set.add(tmp);
            }
        }
        System.out.println(a+b-2*cnt);
    }
}

정리

---

• 문제의 답은 A+B - 2(A∩B)을 구하면 된다.
• Python의 경우에는 set을 이용해 교집합을 구해줬고, Java의 경우에는 성능이 좋은 HashSet을 이용했다.