문제 설명
- 자연수를 원소로 갖는 공집합이 아닌 두 집합 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
예제 출력
4
코드
import java.io.BufferedWriter
import java.io.OutputStreamWriter
import java.util.*
val bw = BufferedWriter(OutputStreamWriter(System.out))
fun main() = with(System.`in`.bufferedReader()){
val (n,m) = readLine().split(" ").map{it.toInt()}
val nSet = readLine().split(" ").map{it.toInt()}.toSet()
val mSet = readLine().split(" ").map{it.toInt()}.toSet()
val nm = nSet.subtract(mSet)
val mn = mSet.subtract(nSet)
bw.write("${nm.size+mn.size}")
bw.flush()
bw.close()
this.close()
}
풀이
대칭 차집합의 개념만 알면 쉽게 풀 수 있는 문제이다 (예전 교육과정엔 있었으나 지금은 있는진 모르겠다..)
두 집합이 있으면, 두 집합의 교집합을 제외한 나머지 부분을 구하는 것이다.
Kotlin Collection Framework 에서는 집합 구조인 Set을 지원하는 데, 여기서 집합 연산을 지원한다.
- Set.intersect - 교집합
- Set.subtract - 차집합
- Set.uniom - 합집합
본 문제 풀이에는 각 집합에서 교집합을 빼서 나온 두 집합(각 집합의 교집합 부분을 제외한 집합)을 합쳐주어 대칭 차집합을 만들어주어 해결하였다.
결과
'PS > BOJ' 카테고리의 다른 글
[BOJ - 22233] [백준 - 22233] 가희와 키워드[Silver - 2] [Solved by Kotlin] (0) | 2024.02.19 |
---|---|
[BOJ - 5397] [백준 - 5397] 키로거[Silver - 2] [Solved by Kotlin] (1) | 2024.01.08 |
[BOJ - 5430] [백준 - 28278] 스택2 [Silver - 4] [Solved by Kotlin] (0) | 2023.12.07 |
[BOJ-17213][백준 17213] 과일 서리 [SILVER-2][Solved by Kotlin] (0) | 2023.10.06 |
[BOJ - 5430] [백준 - 5430] AC [Gold - 5] [Solved by Kotlin] (0) | 2023.08.11 |