[백준] 11723 집합

11723번: 집합

풀이

시간 초과를 피하기 위해 이것저것 다양한 시도를 해봤던 문제이다. 비트 연산으로 바로 해결할 수 있을 줄 알았는데, BufferedReader, BufferedWriter까지 사용해야 정답처리 된다.

코드

import java.io.BufferedReader
import java.io.BufferedWriter
import java.io.InputStreamReader
import java.io.OutputStreamWriter

fun main() {
    var s = 0
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))
    repeat(br.readLine().toInt()) {
        val cmd = br.readLine().split(" ")
        when {
            cmd[0] == "add" -> s = s or (1 shl cmd[1].toInt())
            cmd[0] == "remove" -> s = s and (-1 xor (1 shl cmd[1].toInt()))
            cmd[0] == "check" -> bw.write("${if (s and (1 shl cmd[1].toInt()) != 0) 1 else 0}\n")
            cmd[0] == "toggle" -> s = s xor (1 shl cmd[1].toInt())
            cmd[0] == "all" -> s = -1
            cmd[0] == "empty" -> s = 0
        }
    }
    bw.flush()
}