백준 2948 베르트랑 공준 - Python

출처: 

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

4948번: 베르트랑 공준

def prime_list(n):
    prime = [True] * (n+1)
    
    m = int(n**0.5)
    
    if n <= 3 :
        prime[0] = False
        return [i+1 for i in range(n) if prime[i] == True]
    else:
        for i in range(2,m+1):
            if prime[i] == True:
                for j in range(i+i, n+1, i):
                    prime[j] = False
        return [i for i in range(2,n+1) if prime[i] == True]

while 1:
    a = int(input())
    if a == 0 :
        break
    print(len(prime_list(2*a)) - len(prime_list(a)))

풀이 : 

앞에 에라토스테네스의 체때 만들었던 코드를 그대로 사용했다.

 

리스트로 출력하기 때문에 입력받은 2a의 길이에서 a의 길이를 빼준다.