2051번: 약수 구하기

문제

어떤 자연수 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이라는 숫자가 주어졌을 경우 모든 N의 약수 리스트로 변환
• N의 약수 목록을 작은 순으로 정렬
• N의 약수 중 K번째 작은 약수 찾기
• N의 약수 개수가 K개보다 적을 경우 예외 처리

정답 코드

# p를 1~p까지의 수로 각각 나눈 뒤 나머지가 0인 값(p의 약수인 값)을 모아 리스트로 반환 
def find_divisor(p):
    divisor_list = []
    for i in range(1, p + 1):
        if p % i == 0:
            divisor_list.append(i)
    return divisor_list

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

divisors = find_divisor(n)

# 만일 n의 약수의 개수가 k개보다 적어서 k 번째 약수가 존재하지 않을 경우 처리
if len(divisors) < k:
    print(0)
# n의 약수들 중 k번째로 작은 수를 출력
else:
    print(divisors[k - 1])

채점 결과