BOJ #11723 - 집합 (Python)

문제

비어있는 공집합 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차

import sys ; input = sys.stdin.readline
 
M = int()
S = set()
 
def add(x: int) :
	if _check(x) : return
	S.add(x)
	
def remove(x: int) :
	if not _check(x) : return
	S.remove(x)
 
def check(x: int) :
	print(_check(x))
 
def _check(x: int) :
	if x in S : return 1
	else      : return 0
 
def toggle(x: int) :
	if _check(x) : remove(x)
	else         : add(x)
 
def all() :
	S = set([i + 1 for i in range(20)])
 
def empty() :
	S = set()
 
if __name__ == "__main__" :
	M = int(input().rstrip())
 
	for _ in range(M) :
		command = input()
		
		if len(command.split()) > 1 :
			command, x = command.split()
			x = int(x)
		
		if   'add'    in command : add(x)
		elif 'remove' in command : remove(x)
		elif 'check'  in command : check(x)
		elif 'toggle' in command : toggle(x)
		elif 'all'    in command : all()
		elif 'empty'  in command : empty()

• 왜..틀렸을까..?

2차 - Solved!

import sys ; input = sys.stdin.readline
 
class Set :
	def __init__(self) :
		self.set = set()
		
	def add(self, x: int) :
		if self._check(x) : return
		self.set.add(x)
		
	def remove(self, x: int) :
		if not self._check(x) : return
		self.set.remove(x)
 
	def check(self, x: int) :
		print(self._check(x))
 
	def _check(self, x: int) :
		if x in self.set : return 1
		else             : return 0
 
	def toggle(self, x: int) :
		if self._check(x) : self.remove(x)
		else              : self.add(x)
 
	def all(self) :
		self.set = set([i + 1 for i in range(20)])
 
	def empty(self) :
		self.set = set()
 
if __name__ == "__main__" :
	M = int(input().rstrip())
	S = Set()
 
	for _ in range(M) :
		command = input()
		
		if len(command.split()) > 1 :
			command, x = command.split()
			x = int(x)
		
		if   'add'    in command : S.add(x)
		elif 'remove' in command : S.remove(x)
		elif 'check'  in command : S.check(x)
		elif 'toggle' in command : S.toggle(x)
		elif 'all'    in command : S.all()
		elif 'empty'  in command : S.empty()

• 클래스 구조로 변경하여 해결했다.