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

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

import sys

input = sys.stdin.readline

n, k = map(int, input().split())
cnt = 0
nums = [True] * (n + 1)

for i in range(2, n + 1):
    for j in range(i, n + 1, i):
        if nums[j] == True:
            nums[j] = False
            cnt += 1
            if cnt == k:
                print(j)
                break

 

해설

flag 리스트를 사용해 해결했다. 나머지는 주어진 조건대로 코드를 짜니 해결됐다.