[백준] 에라토스테네스의 체

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

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

 

number 배열을 만들어 차례대로 값을 Fasle 로 바꿔가며 K 번째 해당하는 값을 구하면 된다.

 

import sys, math

N, K = map(int, sys.stdin.readline().split())
answer = 0
number = [True] * (N + 1)

for i in range(2, N + 1):
    for j in range(2, N + 1):
        if number[j]:
            if j % i == 0:
                number[j] = False
                K -= 1
                
                if K == 0:
                    print(j)
                    break
    
    if K == 0:
        break