[백준] 1269 대칭 차집합 JAVA

혅·2024년 3월 19일

1일1백준 -Java-

목록 보기

8/60

문제

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

출력

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

예제 입력

3 5
1 2 4
2 3 4 5 6

예제 출력

내가 했던 풀이 방법

(A-B)와 (B-A)을 각각 계산 후, HashSet에 합친 후 size를 출력 -> 시간 초과 실패 "각 집합의 원소의 개수는 200,000을 넘지 않으며, 모든 원소의 값은 100,000,000을 넘지 않는다." 때문에 for문을 많이 사용하면 안 됨!
A와 B를 합친 후, HashSet으로 변환, HashSet의 size에서 A size만큼을 빼주면, 결과적으로 A-B의 개수임을 이용하여 (A-B)와 (B-A)의 원소 개수를 찾아낸 뒤 합친 결과를 출력.

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;

public class b1269 {
    public static void main(String[] args) throws IOException {

		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] input = br.readLine().split(" ");

        int A = Integer.parseInt(input[0]);
        int B = Integer.parseInt(input[1]);

        String[] setA = br.readLine().split(" ");
        String[] setB = br.readLine().split(" ");

        List<String> array = new ArrayList<>(Arrays.asList(setA));
        array.addAll(Arrays.asList(setB));

        HashSet<String> set = new HashSet<>(array);

        System.out.print(set.size()-A+set.size()-B);
        
    }  
}