개발 연습장/백준 문제풀이

[파이썬, Python] 백준 11723: 집합

LooanCheong 2023. 3. 29. 11:48
반응형

문제

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

 

11723번: 집합

첫째 줄에 수행해야 하는 연산의 수 M (1 ≤ M ≤ 3,000,000)이 주어진다. 둘째 줄부터 M개의 줄에 수행해야 하는 연산이 한 줄에 하나씩 주어진다.

www.acmicpc.net

코드

import sys
input = sys.stdin.readline

m = int(input())
s = set()
li = [i for i in range(1, 21)]

for _ in range(m):
    p = input().split()
    if p[0] == 'add':
        s.add(int(p[1]))
    elif p[0] == 'remove':
        s.discard(int(p[1]))
    elif p[0] == 'check':
        if int(p[1]) in s:
            print(1)
        else:
            print(0)
    elif p[0] == 'toggle':
        if int(p[1]) in s:
            s.remove(int(p[1]))
        else:
            s.add(int(p[1]))
    elif p[0] == 'all':
        s = set(li)
    elif p[0] == 'empty':
        s = set()

설명

주어진 프로그램을 그대로 구현하면 되는 간단한 문제이다.

반응형