글 읽기 - 같은 코드를 PyPy3은 메모리 초과, Python 3은 정답이라고 하네요

yopark 2년 전 1

set()을 쓰든지 비트마스킹을 쓰던지 둘다 짜봤는데 메모리 초과로 계속 안 되길래, PyPy3를 Python 3로 바꿔서 내니까 되네요

PyPy3가 Python 3보다 더 시간을 줄일 수 있다고 들었는데, 메모리는 오히려 더 많이 사용하는 걸까요?

그냥 이제부터 다른 문제도 PyPy3로 내지 말고 Python 3으로 내는게 유리할까요?

### Sol 1. set() 이용

import sys

M = int(sys.stdin.readline().rstrip())
S = set()
for _ in range(M):
    instruction = sys.stdin.readline().rstrip().split()
    if len(instruction) == 1:
        if instruction[0] == 'all':
            S = set([i for i in range(1, 21)])
        else:
            S = set()
        continue
    op, num = instruction[0], instruction[1]
    num = int(num)
    if op == 'add':
        S.add(num)
    elif op == 'remove':
        S.discard(num)
    elif op == 'check':
        print(1 if num in S else 0)
    elif op == 'toggle':
        if num in S:
            S.discard(num)
        else:
            S.add(num)
            
 ### Sol 2. 비트 마스킹 이용

import sys

M = int(sys.stdin.readline().rstrip())
S = [False] * 20

for _ in range(M):
    instruction = sys.stdin.readline().rstrip().split()
    if len(instruction) == 1:
        if instruction[0] == 'all':
            for i in range(20):
                S[i] = True
        else:
            for i in range(20):
                S[i] = False
        continue
    op, num = instruction[0], instruction[1]
    num = int(num) - 1
    if op == 'add':
        S[num] = True
    elif op == 'remove':
        S[num] = False
    elif op == 'check':
        print(1 if S[num] == True else 0)
    elif op == 'toggle':
        S[num] = True if S[num] == False else False