[PS] 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를 공집합으로 바꾼다. 입력 첫째 줄에 수행해야 하는 연산의 수 M (1 ≤ M ≤ 3,000,000)이 주어진다. 둘째 줄부터 M개의 줄에 수행해야 하는 연산이 한 줄에 하나씩 주어진다. 출력 check 연산이 주어질때마다, 결과를 출력한다.

✅ 내 답

- 메모리 초과 방지를 위해 한 줄씩 입력 받아 처리

- 중복을 허용하지 않는 집합(set) 사용

import sys

def check(s, num):
    print(1 if num in s else 0)

def add(s, num):
    s.add(num)

def remove(s, num):
    s.discard(num)

def toggle(s, num):
    if num in s:
        s.remove(num)
    else:
        s.add(num)

if __name__ == "__main__":
    s = set()
    full_set = set(range(1, 21))

    M = int(sys.stdin.readline())
    
    for _ in range(M):
        cmd = sys.stdin.readline().strip().split()
        action = cmd[0]
        
        if action == 'add':
            add(s, int(cmd[1]))
        elif action == 'remove':
            remove(s, int(cmd[1]))
        elif action == 'check':
            check(s, int(cmd[1]))
        elif action == 'toggle':
            toggle(s, int(cmd[1]))
        elif action == 'all':
            s = full_set.copy()
        elif action == 'empty':
            s.clear()

 

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