문제 / K번째 약수/ 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을 출력하시오.

 

▣ 입력예제  6 3

▣ 출력예제  3

---

나의 풀이

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

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


if cnt == []:
    print(-1)
else:
    print(cnt[k-1])

이렇게 풀었는데 오토 채점에서 80점 나왔다...

셀프로 값을 1 5로 넣었는데 

IndexError: list index out of range 에러가 떴다

그래서 또 수정했당

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

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

    if i == k:
        print(cnt[i-1])
        break
        
else:
    print(-1)

첫 수정에는 break를 안걸어줘서

6 3 을 입력하면

 

출력값이

3

-1

로 나왔었다.

그래서 break를 걸어주었다!

 

모범답안

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


else:
    print(-1)

이렇게하면 n의 값이 크더라도 구하고자하는 k번째 수 까지 빠르게 구할 수 있다.

---

※ 알아야 할 것

- for문에도 적절한 break를 걸어주자.