[BOJ] 11723 집합 | Python

문제

백준 11723번 집합 | 실버5

https://www.acmicpc.net/problem/11723

11723번: 집합

• 알고리즘 분류 : 구현 | 비트마스킹

 

 

풀이 과정

파이썬 풀이

import sys
input = sys.stdin.readline

M = int(input())
s = set()
for m in range(M):
    i = input().rstrip().split()
    if len(i) == 1:
        if i[0] == 'all':
            s = set([k for k in range(1, 21)])
        else:
            s = set()
        continue

    if i[0] == 'add':
        s.add(int(i[1]))
    elif i[0] == 'remove':
        s.discard(int(i[1]))
        # remove 대신 discard 사용

    elif i[0] == 'check':
        if int(i[1]) in s:
            print(1)
        else:
            print(0)

    elif i[0] == 'toggle':
        if int(i[1]) in s:
            s.discard(int(i[1]))
        else:
            s.add(int(i[1]))

• M의 최대값이 큰 만큼 시간 초과를 걱정해서 pypy로 제출했는데 메모리 초과 뜸
• 그래서 Python으로 제출했더니 시간 초과
• 참고로 pypy가 python 보다 시간이 짧은 경우가 많지만, 메모리를 더 많이 사용하는 경향이 있음.

 

• all 또는 empty일 때(연산 입력값이 하나일 때) 별도 구분해서 체크
• set에서 remove의 경우 해당 값이 set에 존재하는지 확인 후 존재할 때만 지울 수 있다. 
  만약 존재하지 않는 값을 지우려 할 경우 key error 발생
• remove 대신 discard를 사용하면 값의 존재 여부 확인하지 않고 진행 가능.
  만약 존재하지 않는 값을 입력하더라도 에러가 발생하지 않고 정상적으로 진행