[백준] 4948번 베르트랑 공준 파이썬 코드

코드

import sys
input = sys.stdin.readline

def primenum(n, table):
  table[1] = False # 1은 소수가 아니므로 제외
  for i in range(2, int((2*n)**0.5)+1): # 에라토스테네스의 체 
    if table[i]:
      j = 2
      while i*j <= 2*n:
        table[i*j] = False
        j += 1
  print(table[n+1:2*n+1].count(True)) # n보다 큰 수부터 출력해야한다.

while 1:
  k = int(input())
  if k == 0: break
  numtable = [True]*(2*k+1)
  primenum(k, numtable)