[ 파이썬(python) ] 백준 11723 - 집합

📍 백준 11723 - 집합

백준 11723 - 집합

---

⚡️ 나의 풀이

add와 remove 조건에서 중복된 경우는 무시한다는 조건을 보아 set자료형을 사용했다. 그런데, 정답을 제출하니까 keyError가 떴었다 원인을 찾아보다가 remove 함수 때문이었는데, 값이 없는 경우에 remove를 사용하게 되면 keyError가 난다.

 

이럴 때는 discard함수를 사용하자. 값이 없는 상태에서 discard를 사용해도 에러가 나지 않는다. 또, empty는 빈 집합으로 바꿔야 하는데 그냥 set()으로 초기화해줘도 되지만 clear 함수를 사용해줘도 된다. 시간 복잡도는 크게 차이가 없었다.

 

import sys
input = sys.stdin.readline
 
empty_set = set()
 
def add(x):
    empty_set.add(x)
 
def remove(x):
    empty_set.discard(x)
 
def check(x):
    if x in empty_set:
        return 1
    return 0
 
def toggle(x):
    if x in empty_set:
        empty_set.discard(x)
    else:
        empty_set.add(x)
 
def set_all():
    global empty_set
    empty_set = {i for i in range(1, 21)}
 
def empty():
    empty_set.clear()
 
for _ in range(int(input())):
    command = input().split()
    if command[0] == 'add':
        add(int(command[1]))
    elif command[0] == 'remove':
        remove(int(command[1]))
    elif command[0] == 'check':
        print(check(int(command[1])))
    elif command[0] == 'toggle':
        toggle(int(command[1]))
    elif command[0] == 'all':
        set_all()
    else:
        empty()