본문 바로가기
Beakjoon&프로그래머스/파이썬

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

by 현장 2021. 5. 24.

-코드

from sys import stdin
arr = []
for _ in range(int(input())):
    s = stdin.readline().split()
    if s[0] == "add":
        if int(s[1]) not in arr:
            arr.append(int(s[1]))
    elif s[0] == "remove":
        if int(s[1]) in arr:
            arr.remove(int(s[1]))
    elif s[0] == "check":
        if int(s[1]) in arr:
            print(1)
        else:
            print(0)
    elif s[0] == "toggle":
        if int(s[1]) in arr:
            arr.remove(int(s[1]))
        else:
            arr.append(int(s[1]))
    elif s[0] == "all":
        arr = [i for i in range(1, 21)]
    elif s[0] == "empty":
        arr = []