백준 > 단계별로 풀어보기 > 약수, 배수와 소수 > 2501번: 약수 구하기

문제

어떤 자연수 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번째 약수가 존재하지 않을 경우에는 0을 출력하시오.

예제 입력 1 복사

6 3

예제 출력 1 복사

3

예제 입력 2 복사

25 4

예제 출력 2 복사

0

예제 입력 3 복사

2735 1

예제 출력 3 복사

1

 

오랜만에 답지를 안보고 푼 문제...! 너무 좋다 ㅎㅎ

N, K = map(int, input().split())

lst = []

for i in range(1, N+1):
    if N % i == 0:
        lst.append(i)

if len(lst) < K:
    print(0)

else:
    print(lst[K-1])

 

처음엔 자꾸 원하는 값이 출력되지 않아서 보니 K번째 수는 1부터 시작하고 파이썬의 인덱스는 0부터 시작하니 -1을 해주었다!

 

N을 1부터 N까지 순회하며 나눠준 다음 나머지가 0이면 나누어 떨어진다는 거고 그건 약수라는 소리니 리스트에 추가해준다. 약수의 개수가 K보다 작은 경우는 len함수를 이용해서 체크해주었다.

 

행복하다 ㅎㅎ