[코딩테스트] K번째 약수 - 파이썬

문제

K번째 약수

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

▣ 입력예제 1
6 3

▣ 출력예제 1
3

문제 출처 : 한국정보올림피아드

내가 적용한 코드

import sys
sys.stdin = open("./1번/input.txt", "rt")

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

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

if (len(list) > k):
    print(list[k-1])
else:
    print(-1)

정답 코드

import sys
sys.stdin = open("./1번/input.txt", "rt")

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
else:
    print(-1)

배운 점

• 내가 1차적으로 푼 문제에서는 쓸데없이 list 객체를 통해 약수를 저장하고, 다시 저장된 데이터를 가져오는 행위를 통해 시간이 지연된 것을 확인할 수 있었다.
• 정답 코드를 보게 되면, 반복문 하나를 가지고 if 조건을 통해 한번에 출력을 하려고 시도 하였고, 파이썬에서는 for ~ else 구문이 있어서, 반복문 중에서 break 가 되지 않았다는 것은 k 번째의 약수가 없다는 것이기 때문에 else 구문으로 바로 넘어간다는 사실을 알게 되었다.