1. Problem 📃

📚 출처 - 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 연산이 주어질때마다, 결과를 출력한다.

입출력 예

예제 입력	예제 출력
26 add 1 add 2 check 1 check 2 check 3 remove 2 check 1 check 2 toggle 3 check 1 check 2 check 3 check 4 all check 10 check 20 toggle 10 remove 20 check 10 check 20 empty check 1 toggle 1 check 1 toggle 1 check 1	1 1 0 1 0 1 0 1 0 1 1 0 0 0 1 0

2. Logic 👨‍🏫

Logic
문제에 있는 그대로 코드 작성하면 되는 문제라 바로 코드로 들어가겠습니다.

3. Code 💻

1. 내가 푼 코드 (시간 초과)

import sys
input = sys.stdin.readline

def cal(tmp):
    global result

    if len(tmp) == 1:
        # all
        if tmp[0] == 'all':
            result = set([str(i) for i in range(1, 21)])
        # empty 11
        else:
            result = set()
        return

    else:
        command, target = tmp[0], tmp[1]
        
        # add
        if command == 'add':
            result.add(target)
    
        # check
        elif command == 'check':
            if target in result:
                print(1)
            else:
                print(0)

        # remove
        elif command == 'remove':
            result.discard(target)
    
        # toggle
        elif command == 'toggle':
            if target in result:
                result.discard(target)
            else:
                result.add(target)


M = int(input())
result = set()

for i in range(M):
    tmp = list(map(str, input().split()))
    cal(tmp)

2. 내가 푼 코드 (통과)

import sys
input = sys.stdin.readline
result = set()

M = int(input())

for _ in range(M):
    tmp = input().strip().split()

    if len(tmp) == 1:
        # all
        if tmp[0] == 'all':
            result = set([i for i in range(1, 21)])
        # empty 11
        else:
            result = set()
        continue
    
    command, target = tmp[0], tmp[1]
    target = int(target) # 있고 없고 틀림, 성공 차이가 있음 이유는 왜그런지 모르겠음
    
    # add
    if command == 'add':
        result.add(target)
    
    # check
    elif command == 'check':
        print(1 if target in result else 0)
            
    # remove
    elif command == 'remove':
        result.discard(target) 
    
    # toggle
    elif command == 'toggle':
        if target in result:
            result.discard(target)
        else:
            result.add(target)

4. Feedback 📚

remove vs discard

파이썬 데이터 타입 중에 집합(set) 타입이 있습니다. 집합 타입의 메소드 중에 특정 엘리먼트를 삭제하는 메소드가 두 개 있는데, 각각 remove(), discard()입니다.

s = set()

s.add(10)
s.remove(10)
s.remove(10)

>>>
Traceback (most recent call last):
  File "test.py", line 5, in <module>
    s.remove(10)
KeyError: 10

이미 집합에 아무런 엘리먼트도 없기 때문에 '10'이라는 엘리먼트를 지우려고 할 때 KeyError가 발생한다.

s = set()

s.add(10)
s.discard(10)
s.discard(10)

>>> Process finished with exit code 0

정상적으로 작동합니다.

---

요약해보자면 remove()는 지우려는 엘리먼트가 존재하지 않으면 KeyError를 발생 시키지만 discard()는 엘리먼트가 없어도 정상 종료됩니다.

---

참고 자료 📩
remove vs discard