21-alstjr7437

[alstjr7437]

🔗 문제 링크

집합
그냥 단순한 집합 문제인데 (사실 시간과 몸의 문제로 비트마스킹 이해만 하는 문제로 선정했습니다.)

✔️ 소요된 시간

해결은 10분 3가지 풀이로 하는데 25분 정도..?

📜 전체 문제 풀이

일단 처음에 문제를 선정하면서 구현(set 이용), 비트마스크를 이용해서 풀어봐야겠다고 생각했습니다.

1. 집합으로 문제 풀기
2. 비트마스크 리스트로 문제 풀기
3. 비트마스크로 문제 풀기

1️⃣ 집합

그냥 python에서 집합(set)을 사용해서 구현으로 풀어봤습니다.

set의 add, discard, in 등을 사용해서 집합을 사용했습니다!

import sys

input = sys.stdin.readline

m = int(input())

S = set()

for _ in range(m):
    temp = input().split()
    
    if temp[0] == "add":
        S.add(int(temp[1]))
    if temp[0] == "remove":
        S.discard(int(temp[1]))
    if temp[0] == "check":
        if int(temp[1]) in S:
            print(1)
        else :
            print(0)
    if temp[0] == "toggle":
        if int(temp[1]) in S:
            S.discard(int(temp[1]))
        else:
            S.add(int(temp[1]))
    if temp[0] == "all":
        S = set([i for i in range(1, 21)])
    if temp[0] == "empty":
        S = set()

2️⃣ 리스트를 이용한 비트마스크 사용

주의 : 저가 리스트로 한 방법이 비트마스크가 아닐 수도 있긴한데 저는 비트마스크와 비슷하다고 생각..

일단 집합에 원소가 1~20까지만 있으면 됐습니다.
그래서 20개의 0으로 들어가 있는 리스트를 만들었습니다.
그리고 해당 부분에서 add가 들어오면 해당 자리를 1로 교체하는 방식으로 진행해줬습니다.

import sys

input = sys.stdin.readline

m = int(input())

result = [0] * 21

for _ in range(m):
    cmd = input().split()

    if cmd[0] == "add":
        result[int(cmd[1])] = 1

    if cmd[0] == "remove":
        result[int(cmd[1])] = 0

    if cmd[0] == "check":
        if result[int(cmd[1])] == 1:
            print(1)
        else :
            print(0)

    if cmd[0] == "toggle":
        if result[int(cmd[1])] == 1:
            result[int(cmd[1])] = 0
        else :
            result[int(cmd[1])] = 1

    if cmd[0] == "all":
        result = [1] * len(result)

    if cmd[0] == "empty":
        result = [0] * len(result)

3️⃣ 비트마스크 result만 이용해서

비트마스크 하면 101 -> 5 이런식으로 각 이진수 표현으로 수를 표현하는 자료구조 기법이므로

• add가 들어오면 or 연산자를 이용해서 진행
• remove가 들어오면 and 연산자를 이용해서 진행
• toggle은 xor 연산자를 이용해서 진행
• check는 비트 연산자로 체크하기
• all은 1~20까지 전부 비트가 1로 바뀌도록
• empty는 0으로

import sys

input = sys.stdin.readline

m = int(input())

result = 0

for _ in range(m):
    cmd = input().split()

    if cmd[0] == "add":
        result |= (1 << int(cmd[1]))  # 해당 비트를 1로 설정

    if cmd[0] == "remove":
        result &= ~(1 << int(cmd[1]))  # 해당 비트를 0으로 설정

    if cmd[0] == "check":
        if result & (1 << int(cmd[1])):
            print(1)
        else:
            print(0)

    if cmd[0] == "toggle":
        result ^= (1 << int(cmd[1]))  # 해당 비트를 반전

    if cmd[0] == "all":
        result = (1 << 21) - 1  # 모든 비트를 1로 설정

    if cmd[0] == "empty":
        result = 0  # 모든 비트를 0으로 설정

📚 새롭게 알게된 내용

하나의 문제를 통해서 여러가지 풀이법이 나오고 각 걸리는 시간이 다른걸 확인했습니다.

set 자료구조가 제일 오래 걸렸고 list와 비트마스크를 활용한 방법이 비슷했습니다.(list가 가끔 더빨랐습니다..)

비트연산자 -> 3732
list -> 3632
set 자료구조 -> 5080

[SeongHoonC]

비트마스킹인지 모르고 처음에 세번의 시간초과를 만났는데요..
어떻게든 줄여보려고 hashSet 을 쓰고 했는데 결국 실패했습니다.

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val n = br.readLine().toInt()
    var set = hashSetOf<Int>()

    for (i in 0 until n) {
        val input = br.readLine().split(" ")

        if (input[0] == "all") {
            set = (1..20).toHashSet()
            continue
        }

        if (input[0] == "empty") {
            set.clear()
            continue
        }

        val number = input[1].toInt()
        when (input[0]) {
            "add" -> set.add(number)
            "check" -> if (set.contains(number)) println(1) else println(0)
            "remove" -> set.remove(number)
            "toggle" -> if (!set.remove(number)) set.add(number)
        }
    }
}

kotlin 비트마스킹 잘 몰라서 코드를 참고하면서 만들었습니다.

import java.io.BufferedReader
import java.io.InputStreamReader
import java.lang.StringBuilder

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val n = br.readLine().toInt()

    var bit = 0
    val sb = StringBuilder()

    for (i in 0 until n) {
        val input = br.readLine().split(" ")
        var number = 0

        if (input[0] == "all") {
            bit = bit or 0.inv()
            continue
        }

        if (input[0] == "empty") {
            bit = 0
            continue
        }

        number = input[1].toInt()
        when (input[0]) {
            "add" -> bit = bit or (1 shl number - 1)
            "check" -> {
                val temp = bit and (1 shl number - 1)
                sb.append(if (temp != 0) "1" else "0").append("\n")
            }

            "remove" -> bit = bit and (1 shl number - 1).inv()
            "toggle" -> bit = bit xor (1 shl number - 1)
        }
    }
    println(sb.toString())
}

[wkdghdwns199]

import sys

m = int(sys.stdin.readline())
S = set()

for _ in range(m):
    temp = sys.stdin.readline().strip().split()
    if len(temp) == 1:
        if temp[0] == "all":
            S = set([i for i in range(1, 21)])
        else:
            S = set()
    else:
        func, x = temp[0], temp[1]
        x = int(x)
        if func == "add":
            S.add(x)
        elif func == "remove":
            S.discard(x)
        elif func == "check":
            print(1 if x in S else 0)
        elif func == "toggle":
            if x in S:
                S.discard(x)
            else:
                S.add(x)

[alstjr7437]

import sys

m = int(sys.stdin.readline())
S = set()

for _ in range(m):
    temp = sys.stdin.readline().strip().split()
    if len(temp) == 1:
        if temp[0] == "all":
            S = set([i for i in range(1, 21)])
        else:
            S = set()
    else:
        func, x = temp[0], temp[1]
        x = int(x)
        if func == "add":
            S.add(x)
        elif func == "remove":
            S.discard(x)
        elif func == "check":
            print(1 if x in S else 0)
        elif func == "toggle":
            if x in S:
                S.discard(x)
            else:
                S.add(x)

어허 이양반아 이게 리뷰야?
그냥 본인 코드 올린거잖아

[wkdghdwns199]

비슷하게 나와서 그냥 올렸네요..! 죄송합니다 😢