728x90
문제
비어있는 공집합 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를 공집합으로 바꾼다.
https://www.acmicpc.net/problem/11723
입력
첫째 줄에 수행해야 하는 연산의 수 M (1 ≤ M ≤ 3,000,000)이 주어진다.
둘째 줄부터 M개의 줄에 수행해야 하는 연산이 한 줄에 하나씩 주어진다.
출력
check 연산이 주어질때마다, 결과를 출력한다.
예제
# input
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
# output
1
1
0
1
0
1
0
1
0
1
1
0
0
0
1
0
문제 풀이
import sys
input = sys.stdin.readline
M = int(input())
L = []
for _ in range(M):
command = input().split()
if len(command) == 1:
act = command[0]
else:
act, num = command
num = int(num)
if act == 'add':
if num not in L:
L.append(num)
elif act == 'remove':
if num in L:
L.remove(num)
elif act == 'check':
print(1) if num in L else print(0)
elif act == 'toggle':
if num in L:
L.remove(num)
else:
L.append(num)
elif act == 'all':
L = [i for i in range(1,21)]
else:
L = []
파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.
728x90