11723번 - 집합

시간제한 : 1.5초
메모리 제한 : 4MB

---

문제

비어있는 공집합 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 연산이 주어질때마다, 결과를 출력한다.

---

예제 입력 1

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
1
0
1
0
1
0
1
0
1
1
0
0
0
1
0

---

코드

import sys

m = int(sys.stdin.readline().strip())
S = set()

for _ in range(m):
    commands = sys.stdin.readline().strip().split()

    # num이 없는 경우 -> all, empty
    if len(commands) == 1:
        # all
        if commands[0] == "all":
            S = set(list(range(1, 21)))
        # empty
        else:
            S = set()
    
    # num이 있는 경우
    else:
        command, num = commands[0], commands[1]
        num = int(num)
        
        # add
        if command == "add":
            S.add(num)
        # remove
        elif command == "remove":
            S.discard(num)
        # check
        elif command == "check":
            print(1 if num in S else 0)
        # toggle
        elif command == "toggle":
            if num in S:
                S.discard(num)
            else:
                S.add(num)

---

풀이

시간 : 4000ms
메모리 : 31256KB

이 문제의 경우 S에 num이 있냐 없냐에 따라 명령을 다르게 해야하므로, set을 활용해 간단하게 코드를 작성했다.

먼저 num이 없는 경우와 있는 경우를 나눴다.

• num이 없는 경우

  • all : S를 1 ~ 20까지의 수로 초기화
  • empty : S를 초기화

• num이 있는 경우

  • add : set의 add 명령어를 사용해 S에 num이 이미 있어도 중복을 허용하지 않게하므로 num이 하나만 들어있게 됨
  • remove : set의 discaed 명령어를 사용해 S에 num이 없다고해도 오류가 일어나지 않으므로 있으면 삭제하고 없다면 내버려둠
  • check : if문을 통해 num이 S 안에 있다면 1, 없으면 0을 출력
  • toggle : if문을 통해 num이 S 안에 있다면 num을 S에 discard, 없으면 num을 S에 add