[백준 알고리즘] 2504번 : 대칭 차집합

문제

자연수를 원소로 갖는 공집합이 아닌 두 집합 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을 넘지 않는다.

출력

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

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.StringTokenizer;

public class BOJ_1269 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");

        int A = Integer.parseInt(st.nextToken());
        int B = Integer.parseInt(st.nextToken());

        HashSet<Integer> hashSet = new HashSet<>();

        st = new StringTokenizer(br.readLine(), " ");
        for (int i = 0; i < A; i++)
            hashSet.add(Integer.parseInt(st.nextToken()));

        st = new StringTokenizer(br.readLine(), " ");
        for (int i = 0; i < B; i++) {
            int number = Integer.parseInt(st.nextToken());

            if (hashSet.contains(number)) hashSet.remove(number);
            else hashSet.add(number);
        }
        System.out.println(hashSet.size());
    }
}

풀이 및 느낀점

hashSet을 사용했다.

먼저 A 집합의 원소를 hashSet에 저장했다.
B 집합의 원소를 입력받은 후, hashSet에 저장되어있는지 확인하고 값이 존재한다면 hashSet에서 제거한다. 값이 존재하지 않는다면 hashSet에 B 원소를 저장한다.

저장이 끝나면 hashSet에 남아있는 원소는 대칭 차집합의 원소들이다. hashSet의 크기를 출력하면 정답이다.

참고자료