[백준] java 1269 대칭 차집합

https://www.acmicpc.net/problem/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 의 결과 개수와 B-A의 결과 개수를 합치기만 하면 되는 간단한 문제같다.

Hash 자료구조를 사용하여 중복 값을 쉽게 찾고 대칭 차집합의 원소 개수를 세면 될 것 같다.

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;
import java.util.StringTokenizer;

public class Main {
	public static void main(String[] args) throws Exception{
		BufferedReader br = new BufferedReader( new InputStreamReader( System.in ) );
		StringTokenizer st;
		st = new StringTokenizer( br.readLine() , " ");
		// A집합 개수
		int a = Integer.parseInt(st.nextToken()); 
        // B집합 개수
		int b = Integer.parseInt(st.nextToken());
		// A집합 원소 set, B집합 원소 set
		Set<String> setA = new HashSet<>(), setB = new HashSet<>();
		int countA = 0, countB = 0;
		st = new StringTokenizer(br.readLine(), " ");
        // A집합에 A원소 add, countA ++
		for(int i = 0; i < a; i++ ) {
			setA.add(st.nextToken());countA++;
		}
		st = new StringTokenizer(br.readLine(), " ");
        // B집합에 B원소 넣고 countB ++
		for( int i = 0; i < b; i++ ) {
			String temp = st.nextToken();
			setB.add(temp);countB++;
            // 만약 B원소와 A집합의 원소가 곂치면 countA-- 
            // 결국 countA는 A집합에 대한 차집합 개수 ( A-B )
			if(setA.contains(temp)) {
				countA--;
			}
		}
		
		Iterator<String> i = setA.iterator();
        // 위와 동일함 B-A
		while(i.hasNext()) {
			if(setB.contains( i.next() )) {
				countB--;
			}
		}
		System.out.print(countA+countB);	
	}
}

두 개의 HashSet을 사용하여 차집합의 총 개수를 구하였다.