[Python] 11723 - 집합

풀이

입력된 명령어의 형태를 보면 add, remove, check, toggle의 경우, 문자 + 숫자의 형태이다. 따라서 cmd변수에 입력값을 저장하고, op와 x로 split하여 저장한 후, case에 맞게 나누어 명령어를 실행하도록 구현하였다. (python은 switch-case문이 없어서 if-else문으로 작성해야 한다...)

# 11723 집합

import sys
input = sys.stdin.readline

n = int(input())
s = set()

for _ in range(n):
    cmd = input().strip().split()
    
    if len(cmd) == 1:  # all 또는 empty 명령어
        if cmd[0] == "all":
            s = set(range(1, 21))
        else:  # empty
            s = set()
        continue
        
    op, x = cmd[0], int(cmd[1])  # 명령어와 숫자를 분리
    
    if op == "add":
        s.add(x)
    elif op == "remove":
        s.discard(x)
    elif op == "check":
        print(1 if x in s else 0)
    elif op == "toggle":
        if x in s:
            s.discard(x)
        else:
            s.add(x)