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

1. 문제

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

2. 풀이

약수가 되는 경우 배열에 저장 후 배열 값을 출력해 주었으며, 배열의 index를 초과하는 경우 즉 약수가 존재하지 않거나 약수 개수가 k개보다 적은 경우는 0을 출력하도록 예외처리를 해주었다.

3. 코드

from warnings import catch_warnings

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

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

try:
    print(answer[k-1])
except IndexError:
    print(0)