[파이썬] k번째 약수

K번째 약수

입력설명

첫째 줄에 N과 K가 빈칸을 사이에 두고 주어진다. N은 1 이상 10,000 이하이다. K는 1 이상 N 이하이다.

출력설명

첫째 줄에 N의 약수들 중 K번째로 작은 수를 출력한다. 만일 N의 약수의 개수가 K개보다 적어서 K번째 약수가 존재하지 않을 경우에는 -1을 출력하시오.

입력예제

1 6 3

출력예제
1 3

🥳 나의 답 (채점 결과 100)

import sys
#sys.stdin=open("input.txt","rt")

n,k=map(int,input('').split())
d=[]

for i in range (1,n+1):
	# 나머지가 0이면(=약수) 리스트에 추가
	if n%i==0:
    	#약수 추가
        d.append(i)
# 약수 개수가 k보다 작으면 -1출력
if len(d)<k:
    print(-1)
#이외에는 리스트의 k-1인덱스 값 출력
else: print(d[k-1])

✔ 정답 풀이

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

for i in range(1,n+1):
    #약수면 카운트 +1
    if n%i==0:
        cnt+=1
    # 카운트 = k 이면 i를 출력하고 종료
    if cnt==k:
        print(i)
        break

else: #break 하지 않은 경우 실행
    print(-1)

 

🤷‍♀️ 의견

리스트로 하지 않아도 풀수 있구나!

정답이 더 간단하게 할 수 있으니 참고하자!

(내가 너무 리스트에 얽매이기는 하는듯..)