[백준/Java] 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을 넘지 않는다.

 

| 출력

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

 

| 풀이

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
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());

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

        HashSet<Integer> setA = new HashSet<>();
        HashSet<Integer> setB = new HashSet<>();

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

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

        int count = 0;

        for(int num : setA)
        {
            if(!setB.contains(num))
            {
                count++;
            }
        }

        for(int num : setB)
        {
            if(!setA.contains(num))
            {
                count++;
            }
        }
        System.out.println(count);
    }
}

 

| 정리

첫 풀이는 ArrayList를 활용해서 풀었는데 시간 초과가 떴다.

HashSet이 더 빠르다는 것은 알고 있었지만, 조건 중에서 중복을 허용하지 않는다 라는 조건이 없어서 일부러 HashSet을 사용하지 않았다.

시간 초과가 뜨고 생각을 해보니,,, 차집합의 갯수를 세는 거니까 중복을 허용하지 않아도 문제 될 것이 없다,,? 라는 결론이 났다. 왜냐면 차집합을 구하려면 해당 원소가 있는지 없는지만 판별하면 되는 거고 중복되는 원소가 몇개 있는지는 중요하지 않다. 라는 생각에 HashSet으로 수정해서 다시 풀었고

맞았습니다!를 만났다~!