글 읽기 - 파이썬) 더 나은 풀이가 있을까요?

hg7258 2년 전 1

check나 toggle때

in x in arr~~ 식의 풀이는 O(n)이라서, 나름 배열 활용해서 했는데 시간을 엄청 간당간당하게 남기고 통과했더라고요.

시간을 더 효율적으로 쓸 방법이 있을까요?

import sys
input=sys.stdin.readline    
arr=set()
dp=[0] * 21
for _ in range(int(input())):
    inp=[x for x in input().split()]
    a=inp[0]
    try:
        b=inp[1]
    except:
        pass    
    if a=="add":
        arr.add(int(b))
        dp[int(b)]=1
    elif a=="remove":
        if dp[int(b)]==1:
            arr.remove(int(b))
        dp[int(b)]=0
    elif a=="check":
        if dp[int(b)]==1:
            print(1)
        else:
            print(0)
    elif a=="toggle":
        if dp[int(b)]==1:
            dp[int(b)]=0
            arr.remove(int(b))
        else:
            dp[int(b)]=1
            arr.add(int(b))
    elif a=="all":  
        arr=set(range(1,21))
        dp=[1] * 21
    elif a=="empty":
        arr.clear()
        dp=[0] * 21

recoma 2년 전 0

x의 범위가 작기 때문에 배열 대신 비트마스크를 이용하면 더 빠르게 통과할 수 있습니다.

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

파이썬) 더 나은 풀이가 있을까요?

hg7258 2년 전 1

recoma 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말