Python | K번째 약수

K번째 약수

문제 🧐

어떤 자연수 p와 q가 있을 때, 만일 p를 q로 나누었을 때 나머지가 0이면 q는 p의 약수이다.
6을 예로 들면
6 ÷ 1 = 6 … 0
6 ÷ 2 = 3 … 0
6 ÷ 3 = 2 … 0
6 ÷ 4 = 1 … 2
6 ÷ 5 = 1 … 1
6 ÷ 6 = 1 … 0
그래서 6의 약수는 1, 2, 3, 6, 총 네 개이다.
두 개의 자연수 N과 K가 주어졌을 때, N의 약수들 중 K번째로 작은 수를 출력하는 프로그램을
작성하시오.

입력

첫째 줄에 N과 K가 빈칸을 사이에 두고 주어진다. N은 1 이상 10,000 이하이다. K는 1 이상 N
이하이다.

6 3

출력

첫째 줄에 N의 약수들 중 K번째로 작은 수를 출력한다. 만일 N의 약수의 개수가 K개보다 적어서 K번째 약수가 존재하지 않을 경우에는 -1을 출력하시오

3

코드 💻

#내 풀이
n,k=map(int, input().split())

num=[]
cnt=0 #약수의 갯수를 세는 변수 
for i in range(1,n+1):
    if n % i == 0: #i가 n의 약수이면 나머지가 0으로 나누어 떨어지므로(i가 n의 약수인지 판별)
        num.append(i) #num 리스트에 약수 i 저장
        cnt += 1 #약수일 때 1증가 

if cnt >= k: 
    print(num[k - 1])
else:
    print(-1)

출처 : 한국정보올림피아드