[Algorithm][Python] K번째 약수

[문제]  K번째 약수 

어떤 자연수 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번째 약수가 존재하지 않을 경우에는 -1을 출력하시오.

 

입력예제 1

6 3 

 

출력예제 1

3

 

 

문제 접근

1. N을 for문으로 1부터 N+1까지 돌린다.
2. N % i 나머지 연산을 했을 때, 0이면 약수이니 cnt 변수로 +1을 복합 연산을 해준다.
3. N의 약수중에 K번째 약수가 없다면 -1을 출력한다. 

 

풀이

 

나의 코드

• 14 행 : if 문을 통해 K가 cnt보다 크면 N의 약수는 존재하지 않기에 -1을 출력해줍니다.

N,K = map(int, input().split())
cnt = 0

for i in range(1, N+1): 

    if N % i == 0:
        cnt += 1

    if cnt == K: # K번째 약수라면
        print(i)
        break
    
# cnt == k가 안걸릴 때, 즉 약수의 갯수(n)이 k보다 작으면
if K > cnt:
    print(-1)
    

 

정답 코드

• for - else 구문을 사용! 
  • for문이 중간에 break 등으로 끊기지 않고, 끝까지 수행되면 else 구문이 수행되는것이 for-else구문입니다.

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

for i in range(1, n+1):
    if n % i == 0:
        cnt += 1

    if cnt == k: 
        print(i)
        break

# for else구문
# for문의 break가 안걸리면 else구문 실행
# cnt == k가 안걸릴 때, 즉 약수의 갯수(n)이 k보다 작으면
else:
    print(-1)