set() 자료 구조를 사용한다
import sys
sys.stdin = open('input.txt')
N = int(input())
sset = set()
for i in range(N):
data = input()
if 'add' in data:
sset.add(int(data.split()[1]))
elif 'remove' in data:
if int(data.split()[1]) in sset:
sset.remove(int(data.split()[1]))
elif 'check' in data:
if int(data.split()[1]) in sset:
print(1)
else:
print(0)
elif 'toggle' in data:
if int(data.split()[1]) in sset:
sset.remove(int(data.split()[1]))
else:
sset.add(int(data.split()[1]))
elif 'all' in data:
sset = set((range(1,21)))
elif 'empty' in data:
sset = set()
파이썬에는 set()이라는 집합을 처리할 수 있는 자료구조가 있어서 쉽게 해결이 가능하다.
대부분의 연산을 set을 통해 해결 할 수 있다. (파이썬 충성!)