[python] 백준 11723 집합

https://www.acmicpc.net/problem/11723

11723번: 집합

import sys

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

for i in range(n):
    inp = sys.stdin.readline().strip().split()
    
    if len(inp) == 1:
        if inp[0] == 'all':
            s = set(range(1, 21))
        else:
            s = set()
        continue
        
    command, num = inp[0], int(inp[1])
    
    if command == 'add':
        s.add(num)
    if command == 'remove':
        s.discard(num)
    if command == 'check':
        print(1 if num in s else 0)
    if command == 'toggle':
        if num in s:
            s.discard(num)
        else:
            s.add(num)