[BOJ] 1269. 대칭 차집합

문제

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

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

해결 방법

• a 집합과 b 집합을 각각 hashmap에 저장
• a-b 인 경우 acount++
• b-a 인 경우 bcount++
• acount+bcount
• hashmap의 keyset을 사용하면 반복문을 돌릴 수 있다.

  // 대칭 차집합
  public class Week04_1269 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int a = Integer.parseInt(st.nextToken());
        int b = Integer.parseInt(st.nextToken());
        st = new StringTokenizer(br.readLine());
        HashMap<Integer, Integer> ahm = new HashMap<>();
        for (int i = 0; i < a; i++) {
            ahm.put(Integer.parseInt(st.nextToken()), i);
        }
        st = new StringTokenizer(br.readLine());
        HashMap<Integer, Integer> bhm = new HashMap<>();
        for (int i = 0; i < b; i++) {
            bhm.put(Integer.parseInt(st.nextToken()), i);
        }
        int aMinusB = 0;
        for (int i : ahm.keySet()) {
            if (!bhm.containsKey(i)) {
                aMinusB++;
            }
        }
        int bMinusA = 0;
        for (int i : bhm.keySet()) {
            if (!ahm.containsKey(i)) {
                bMinusA++;
            }
        }
        bw.write(aMinusB + bMinusA + "");
        bw.flush();
        bw.close();
    }
  }