m1ndy5's coding blog

[수학] 백준 4948번 베르트랑 공준 with Python(pypy3와 python3의 차이점은?) 본문

알고리즘 with python/수학

[수학] 백준 4948번 베르트랑 공준 with Python(pypy3와 python3의 차이점은?)

정민됴 2023. 2. 27. 21:16

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

import math

def cnt_prime_num(n):
    lst = [1] * ((2*n)+1)
    lst[0] = 0
    lst[1] = 0
    for i in range(2, int(math.sqrt(2*n))+1):
        j = 2
        while i*j <= 2*n:
            if lst[i*j] == 1:
                lst[i*j] = 0
            j += 1
    lst = lst[n+1:(2*n)+1]
    return lst.count(1)

n = int(input())
while n != 0:
    print(cnt_prime_num(n))
    n = int(input())

문제다 문제!!! python3로 시간초과나서 혹시? 하고 pypy3 했더니 통과가 됐다.
둘의 차이점은?!
pypy3는 python3 보다 복잡한 코드에서 속도가 빠르다는 장점이 있다.
이유는? 같은 코드가 반복되면 그 반복되는 코드를 캐싱(저장)하기 때문에 시간이 더 빠르다.
오잉 그렇다면 pypy3을 쓰는게 python3보다 이득 아님? 할 수도 있겠지만 반대로 python3가 pypy3보다 간단한 코드에서는 메모리 사용량과 시간 모두 더 빠르다고 한다.
하지만 python3으로도 통과한 사람들이 있어서 pypy3으로는 통과하긴 했지만 python3도 통과할만한 코드를 생각했고 (물론 조금 다른 사람 코드 참고했다ㅎ)

import math

def prime_num():
    lst = [1] * ((123456*2)+1)
    lst[0] = 0
    lst[1] = 0
    for i in range(2, int(math.sqrt(2*123456))+1):
        j = 2
        while i*j <= 2*123456:
            if lst[i*j] == 1:
                lst[i*j] = 0
            j += 1
    return lst

n = int(input())
lst = prime_num()
while n != 0:
    temp = lst[n+1:(2*n)+1]
    print(temp.count(1))
    n = int(input())

생각해보니 n은 1부터 123456이라는 범위가 지정되어 있었고 미리 123456까지의 소수를 구해놓는다면 반복문이 돌아갈 때마다 구할 필요가 없었다!!! 그리고 math.sqrt()를 사용하지 않고 ** 0.5 를 사용해도 됐다!!ㅎㅎㅎ 굳!!