백준 2501번 약수 구하기(python)

1. 문제

https://www.acmicpc.net/problem/2501

2. 풀이

약수가 되는 값을 배열에 저장 후 k보다 배열 크기가 작은 경우가 아니라면 index가 k - 1인 값을 출력한다.

3. 코드

n,k = map(int,input().split())
arr = []
for i in range(1,n + 1):
    if n % i == 0:
        arr.append(i)

if len(arr) >= k:
    print(arr[k - 1])
else:
    print(0)