Python Algorithm) 1. k번째 약수

inflearn의 파이썬 알고리즘 문제풀이 를 시작했습니다. 

1. 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 이하이다.

 

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

 

▣ 입력예제 1

6 3

▣ 출력예제 1

3

 

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

[자료구조와 알고리즘 입문]

#input으로 주어지는 것은 문자열이기때문에 숫자형태로 변환 필요
N, k = map(int, input().split())

#처음에는 일단 약수를 list 형태로 다 구한 다음, k-1의 인덱스의 값을 출력하려고 했다. 
#그러나 이 방법으로는 N이 많이 커지면 k번째 약수를 구했을 지라도 계속 반복해야 하므로 비효율적이라는 생각이 들었다. 
#그래서 cnt로 몇 번째 약수인지 구해주고, 그게 k가 되면 출력하고 멈추기로 했다. 이 과정에서
#k가 총 약수의 개수보다 큰 경우 -1을 출력해야 하므로 for else 구문을 쓰기로 했다. 


cnt = 0
for i in range(1, N+1):
	if N%i == 0:
    	cnt += 1
    if cnt == k:
    	print(i)
        break
        
else: #for문 뒤에 오는 else는 'for문이 다 돌았다면'이란 의미이다 
	print(-1)