[구현력 기르기]-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 63

▣ 출력예제 1 3

 

나의코드

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

res=[]

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

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

 문제에서 n의 약수들중 k 번째로 작은 수가 있으면 그 수를 출력하고 없으면 -1을 출력 하는 문제이다.

res라는 n의 약수를 담을 리스트를 선언한 이후 반복문을 사용해서 약수를 찾을 때 마다 res에 추가하였다.

그 후 최종적으로 K번째 수가 존재하느지 안하는지를 확인하는 if문을 사용하였다.

 

 

정답코드

import sys
sys.stdin=open("input.txt", "r")
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부터 순회 하는 과정에서 자동으로 K에 대한 인덱싱이 정해지는 법을 사용한 점은 비슷하다.!

하지만 직관적으로는 나의 코드가 더 이해하기 좋다고 생각된다.