코딩/백준

[백준/파이썬/수학] 2501번 약수 구하기 코딩테스트 연습풀이 브론즈3

thisisjade 2022. 3. 28. 21:25
728x90

문제 출처: https://www.acmicpc.net/problem/2501

 

2501번: 약수 구하기

첫째 줄에 N과 K가 빈칸을 사이에 두고 주어진다. N은 1 이상 10,000 이하이다. K는 1 이상 N 이하이다.

www.acmicpc.net

 

문제

어떤 자연수 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번째로 작은 수를 출력하는 프로그램을 작성하시오.

 

입력

첫째 줄에 N과 K가 빈칸을 사이에 두고 주어진다. N은 1 이상 10,000 이하이다. K는 1 이상 N 이하이다.

 

출력

첫째 줄에 N의 약수들 중 K번째로 작은 수를 출력한다. 만일 N의 약수의 개수가 K개보다 적어서 K번째 약수가 존재하지 않을 경우에는 0을 출력하시오.

 

예제 입력 1

6 3

 

예제 출력 1

3

 

예제 입력 2

25 4

 

예제 출력 2

0

 

예제 입력 3

2735 1

 

예제 출력 3

1

 

정답

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

for i in range(1, n+1):
    if n % i == 0:
        tmp+=1
    if k == tmp:
        print(i)
        break
    
    if n == i:
        print(0)

1. n, k 값 입력, tmp 0으로 세팅

2. for문에서 range(1, n+1)로 세팅하여 1부터 n까지로 범위 지정

3. 첫번째 if문 n의 약수 일경우 tmp +1

4. 두번째 if문 k와 tmp의 값이 같을때 현재 숫자인 print(i)

5. 두번째 if문에서 걸러지지 않으면 0으로 세팅이기 때문에 세번째 if문에서 n과 i가 같은 경우 print(0)

728x90