[백준] 2501 약수 구하기 Python

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

 

📝  문제

어떤 자연수 p와 q가 있을 때, 만일 p를 q로 나누었을 때 나머지가 0이면 q는 p의 약수이다. 
6을 예로 들면
6 ÷ 1 = 6 … 0
6 ÷ 2 = 3 … 0
6 ÷ 3 = 2 … 0
6 ÷ 4 = 1 … 2
6 ÷ 5 = 1 … 1
6 ÷ 6 = 1 … 0
그래서 6의 약수는 1, 2, 3, 6, 총 네 개이다.
두 개의 자연수 N과 K가 주어졌을 때, N의 약수들 중 K번째로 작은 수를 출력하는 프로그램을 작성하시오.

 

🔎  풀이

약수의 개수를 저장 할 count라는 변수와 약수의 개수가 K보다 작은지를 비교하기 위한 변수 check을 생성한다.

1부터 N까지 나눗셈 연산을 진행한다.

연산을 수행하는 중 나머지가 0이면 count의 값을 1 증가시키고, count의 값이 K와 동일할 때 check = True로 값을 변경한 뒤 반복문을 탈출한다.

check 변수의 값에 따라 약수 또는 0을 출력한다.

 

💻  코드

N, K = map(int, input().split())
count = 0
check = False
for i in range(1, N + 1):
    if N % i == 0:
        count += 1
    if count == K:
        check = True
        break
if check:
    print(i)
else:
    print(0)