[Python] 백준 2501: 약수 구하기 파이썬

백준 2501번

 

 

풀이 : 

약수 구하는 함수를 먼저 만들기

제곱근의 약수까지만 구하면 되는 빠른 방법 이용

 

코드:

import sys

input = sys.stdin.readline

def yaksu(n):
    div = []

    for i in range(1, int(n**(1/2))+1):
        if (n % i == 0):
            div.append(i)
            if ((i**2) != n):
                div.append(n//i)

    div.sort()
    return div

n, k = map(int, input().rstrip().split())

div = yaksu(n)

if (len(div) < k):
    print(0)
else:
    print(div[k-1])

 

** 약수 구하는 부분 잘보기