[백준/11723] 집합

💻 전체 코드

import sys
n = int(sys.stdin.readline())
s = set()

for _ in range(n):
    words = sys.stdin.readline().split()
    command = words[0]
    
    if command == 'add':
        s.add(int(words[1]))
    elif command == 'remove':
        s.discard(int(words[1]))
    elif command == 'check':
        if int(words[1]) in s:
            print(1)
        else:
            print(0)
    elif command == 'toggle':
        if int(words[1]) in s:
            s.discard(int(words[1]))
        else:
            s.add(int(words[1]))
    elif command == 'all':
        s = set(range(1,21))
    elif command == 'empty':
        s = set()

 

집합 자료형에 대해서 잘 모르는 상태였어서 서치를 하면서 풀다가 공간/시간 복잡도를 줄일 수 있ㄴ는 만큼 줄였다.

연속된 수는 항상 range(n,m)이런식으로 만들었었는데, 틀렸다고 하길래 프린트를 해보니까,,

 

단순히 range(n,m)이 아닌 list(range(n,m))이라든지, set(range(n,m)) 이렇게 사용을 해야 한다.

집합을 표현할 때 보통 set([1,2,3]) / set({1,2,3}) / {1} 이런식으로 선언하는데 range는 set(range(n,m))으로 선언한다는 점,,

for문을 사용해서 연속된 수를 표현해도 되지만, range를 사용하는 것이 훨씬 효율적이다.

 

다른 분들은 int(words[1]) 표현을 직접 data 등으로 선언해서 구현하셨는데,

그렇게 되면 입력이 다를 때를 대비해서 if - else 문을 한번 더 구현해야 한다.

나는 if - else 문을 한번 더 쓰는 것보다 가독성이 떨어지는 쪽을 선택했다. 컴퓨터가 나보다 더 편해야하니까..?🐤

조건문을 많이 사용하면 파이프라이닝 시에 골치아픈 경우가 생긴다.(고 컴퓨터구조론 시간에 배웠다!)

 

set 자료형과 자잘한 팁에 대해서 알게된 문제이다.