[백준 / 파이썬] 11723번 집합

import sys

n = int(sys.stdin.readline())
a = set()

for i in range(n):
    k = sys.stdin.readline().strip().split()

    if k[0] == 'add':
        a.add(int(k[1]))
    elif k[0] == 'remove':
        a.discard(int(k[1]))
    elif k[0] == 'check':
        if int(k[1]) in a:
            print(1)
        else:
            print(0)
    elif k[0] == 'toggle':
        if int(k[1]) in a:
            a.discard(int(k[1]))
        else:
            a.add(int(k[1]))
    elif k[0] == 'all':
        a = set([i for i in range(1, 21)])
    else:
        a = set()

 

집합을 이용하는 문제이기 때문에 set을 사용한다

remove 대신 discard를 사용했는데 remove 함수는 존재하지 않는 수를 제거하려고 했을 때 오류가 발생한다