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

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

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

 

힘든 화요일을 위해 

에라스토스테네스의 체 복기

 

카운트로 개수를 세주고 같으면 출력

n, k = map(int, input().split())

arr = [1] * (n+1)
cnt = 0

for i in range(2, n+1):
    if arr[i] == 1: #아직 안지웠으면
        for j in range(i, n+1, i):
            if arr[j] == 1:
                cnt+=1
            
            if cnt == k:
                print(j)
                break
            arr[j]+=1 #i씩 증가시키면서 지운다.

 

완성