[백준] 11723 집합

Python Code

import sys
M = int(sys.stdin.readline())
S = set()
for _ in range(M):
    command = sys.stdin.readline().strip().split()

    if len(command) == 1:
        if command[0] == 'all':
            S = set([i for i in range(1, 21)])
        else:
            S = set()

    else:
        func, num = command[0], command[1]
        num = int(num)

        if func == 'add':
            S.add(num)
        elif func == 'remove':
            S.discard(num)
        elif func == 'check':
            print(1 if num in S else 0)
        elif func == 'toggle':
            if num in S:
                S.discard(num)
            else:
                S.add(num)