[알고리즘] K번째 약수

(문제) K번째 약수


어떤 자연수 p와 q가 있을 때, 만일 p를 q로 나누었을 때 나머지가 0이면 q는 p의 약수이다.

6을 예로 들면

그래서 6의 약수는 1, 2, 3, 6, 총 네 개이다.

두 개의 자연수 N과 K가 주어졌을 때, N의 약수들 중 K번째로 작은 수를 출력하는 프로그램을 작성하시오.

입력설명

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

K는 1 이상 N 이하이다.

출력설명

첫째 줄에 N의 약수들 중 K번째로 작은 수를 출력한다.

만일 N의 약수의 개수가 K개보다 적어서 K번째 약수가 존재하지 않을 경우에는 -1을 출력하시오.

테스트케이스

입력예제 출력예제
6 3 3
8 4 8
25 5 -1
100 5 10
100 7 25
1000 12 125

해결방법

  1. K번째로 작은 약수를 출력하는 프로그램이므로 K번째를 나타낼 수 있는 count라는 변수를 생성
  2. 반복문을 1 부터 N의 숫자까지 돌려서 약수가 되는 지점에 count 변수를 1 씩 증가
  3. count가 설정한 k의 값과 동일해지면 반복문을 종료 후, 반복문을 통한 i 값을 출력
  4. 만약 k번째 약수가 없을 경우 -1 출력한다

코드

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

count = 0

for i in range(1, n+1):
    if n % i == 0:
        count = count + 1

    if count == k:
        print(i)
        break
else:
    print(-1)

댓글남기기