백준 2501 약수 구하기

내가 쓴 정답코드

import sys
n, k = map(int, sys.stdin.readline().split())
# print(n, k) 출력값 n k

li = []


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

if len(li) >= k:  # k번째 약수가 존재하면 출력
    print(li[k - 1])
else :
    print(0)