[백준] 2960 : 에라토스테네스의 체 - Python

www.acmicpc.net/problem/2960

2960번: 에라토스테네스의 체

import sys, math

N, K = map(int, sys.stdin.readline().split())
num = [1] * (N+1)
answer = 0
for i in range(2, N+1):
  for j in range(i, N+1, i):
    if num[j]:
      num[j] = 0
      answer += 1
      if answer == K:
        print(j)
        break