[백준] 2501번 약수 구하기 | 파이썬 Python

문제

어떤 자연수 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을 출력하시오.

 

예제

 

나의 풀이

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

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

if cnt != K:
    print(0)

• k번째 약수에 도달하면 for문을 종료하고, 찾지 못하면 `0`을 출력한다.

 

다른 사람의 풀이

N, K = map(int, input().split())
num_list = []

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

if len(num_list) >= K:
    print(num_list[K-1])
else:
    print(0)

• 리스트의 길이를 이용해서 답을 출력한다.

 

새롭게 깨달은 점

위 리스트 사용, 아래 cnt 사용

나는 나의 풀이가 `break`로 for문을 일찍 종료하기 때문에 속도가 더 빠를 거라고 생각했다. 하지만 결과적으로는 리스트를 사용한 다른 사람의 풀이가 속도가 더 빨랐다.

 

어차피 `K`가 `N`에 가깝다면 여전히 거의 모든 경우를 실행하게 되고, 거기에 for문에 if문이 두 개를 처리하느라 실행 시간이 더 늘어난 거 같다. 앞으로는 여러 경우의 수를 떠올리며 더욱 효율적인 풀이를 작성해야겠다.