[백준/BOJ] 1269번: 대칭 차집합

https://www.acmicpc.net/problem/1269

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개이다.

해설

처음엔 contains 함수를 이용하여 각각 개수를 세어 더했지만 시간초과가 나왔다.

다음으론 집합 Set을 이용하여 문제를 해결하였다.

Set은 중복이 되지 않는다는 특성이 있다.

A, B의 원소를 모두 Set에 넣은 후 Set의 크기에 B의 개수를 빼면 A-B와 같고, A의 개수를 빼면 B-A의 결과와 같다.

따라서 Set.size - n(A의 개수) + Set.size - m(B의 개수)를 해주면 A와 B의 대칭 합집합의 원소의 개수를 구할 수 있다.

소스 코드

import java.util.*
import kotlin.collections.HashSet

fun main() {
    var st = StringTokenizer(readLine())
    val n = st.nextToken().toInt()
    val m = st.nextToken().toInt()
    val set = HashSet<Int>()
    st = StringTokenizer(readLine())
    for(i in 0 until n) set.add(st.nextToken().toInt())
    st = StringTokenizer(readLine())
    for(i in 0 until m) set.add(st.nextToken().toInt())
    println(set.size-n + set.size-m)
}