[백준/Python] 2501번: 약수 구하기

단계별로 풀어보기 > 단계 9: 약수, 배수와 소수 > 2501번

#2501: 약수 구하기
n, k = map(int, input().split())
divisors = []
for i in range(1,n+1):
  if n%i==0:
    divisors.append(i)
if k>len(divisors):
  print(0)
else:
  print(divisors[k-1])

 

n을 1부터 n까지의 수로 나누었을 때 나머지가 0인 i들만 약수 리스트에 저장합니다. 출력 시 k번째 항목을 출력하는데, 이때 k가 리스트의 길이보다 길면 오류가 뜨므로 이 경우 0을 출력하도록 설정해줍니다.