[python] 백준 11723 집합

 

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

11723번: 집합

 

문제 해결

집합의 연산에 대해서 확인 할 수 있는 기본적 문제라 생략.

 

CODE

import sys
input = sys.stdin.readline

M = int(input())
S = set()
for _ in range(M):
    command = input().split()
    if command[0] == 'add':
        S.add(int(command[1]))
    elif command[0] == 'remove':
        try:
            S.remove(int(command[1]))
        except:
            continue
    elif command[0] == 'check':
        if int(command[1]) in S:
            print(1)
        else:
            print(0)
    elif command[0] == 'toggle':
        if int(command[1]) in S:
            S.remove(int(command[1]))
        else:
            S.add(int(command[1]))
    elif command[0] == 'all':
        S = set([i for i in range(1,21)])
    else:
        S = set()