[Python] 백준 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())
ans = []

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

ans.sort()  # 이미 값은 sort 되어 있으므로 굳이 필요x

if len(ans) < k:
    print(0)
else:
    print(ans[k-1])

 

 

다른 코드 이해

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

result = 0

for i in range(1, N + 1):
    if N % i == 0:
        K -= 1
        if K == 0:
            result = i
            break

print(result)

1부터 N까지 순서로 약수를 구한다.
K번째 약수가 될때까지 -1 하고, K가 0이 되면 이때 약수값을 출력한다.

 

ex)

N = 6, K = 3

 

1) i = 1    ->   K = 2

2) i = 2    ->   K = 1

3) i = 3    ->   K = 0

 

따라서, 정답은 3

 

 

 

 

 

https://www.acmicpc.net/problem/2501

2501번: 약수 구하기