[BOJ 2501] 약수 구하기

약수 구하기 문제

나의 코드

import sys
N, K = map(int, input().split())
result = []

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

result.sort()

if len(result) < K :
    print(0)
else: 
    print(result[K-1])

참고하기 좋은 코드

import sys
N, K = map(int, input().split())
result = 0

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

else : 
    print(-1)