BAEKJOON_2501 "약수 구하기" PYTHON

[백준] 2501번 Python 파이썬

 

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

2501번: 약수 구하기

n, k = map(int, input().split())
count = 0

for i in range(1, n+1):
    if n % i == 0:
        count += 1 # i로 n을 나눴을 때, 나누어 떨어질 때마다 count에 1씩 더해준다.
    if count == k:
        a = i # count가 k와 같아진다면 출력할 a에 i를 넣어준다.
        break
    else:
        a = 0 # 위의 조건들에 해당하지 않는 경우 0을 출력해야하기 때문에 a에 0을 넣어준다.
print(a)