[백준] 2960번 에라토스테네스의 체 - 파이썬

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

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

풀이

에라토스테네스의 체 알고리즘을 알고있다면 변형하여 무난하게 풀 수가 있다.

에라토스테네스의 체 알고리즘 설명: https://sunghyun98.tistory.com/33?category=950313

# 2960번 에라토스테네스의 체
n, k = map(int, input().split())
arr = [True for _ in range(n+1)]
def findsosu():
    cnt = 0
    for i in range(2, n+1):
        if arr[i]:
            for j in range(i, n+1, i):
                if arr[j] == True:
                    arr[j] = False
                    cnt+=1
                    if cnt == k:
                        return print(j)
                
findsosu()

출력결과