백준 11723 집합 파이썬

문제

비어있는 공집합 S가 주어졌을 때, 아래 연산을 수행하는 프로그램을 작성하시오.

add x: S에 x를 추가한다. (1 ≤ x ≤ 20) S에 x가 이미 있는 경우에는 연산을 무시한다.
remove x: S에서 x를 제거한다. (1 ≤ x ≤ 20) S에 x가 없는 경우에는 연산을 무시한다.
check x: S에 x가 있으면 1을, 없으면 0을 출력한다. (1 ≤ x ≤ 20)
toggle x: S에 x가 있으면 x를 제거하고, 없으면 x를 추가한다. (1 ≤ x ≤ 20)
all: S를 {1, 2, ..., 20} 으로 바꾼다.
empty: S를 공집합으로 바꾼다.

풀이

주어진 조건대로 해결하면 되는 문제!

처음의 오답이유:
all이 나올때 S = {1,2,3...,20}으로 정수를 저장했었다.
보다시피 입력받을때의 value는 str이다. 그래서 정수 !=문자열이여서 인식을 못했던것을 뒤늦게 깨달았다. 타입을 잘 통일하자!

import sys
input = sys.stdin.readline

t = int(input())
S = []
for _ in range(t):
    value = list(input().split())
    if value[0] == 'add':
        if value[1] in S:
            pass
        else:
            if type(S) == list:
                S.append(value[1])
            else:
                S.add(value[1])
    elif value[0] == 'remove':
        if value[1] not in S:
            pass
        else:
            S.remove(value[1])
    elif value[0] == 'check':
        if value[1] in S:
            print(1)
        else:
            print(0)
    elif value[0] == 'toggle':
        if value[1] in S:
            S.remove(value[1])
        else:
            if type(S) == list:
                S.append(value[1])
            else:
                S.add(value[1])
    elif value[0] == 'all':
        S = {'1','2','3','4','5','6','7','8','9','10','11','12','13','14','15','16','17','18','19','20'}
    elif value[0] == 'empty':
        S = []