파이썬 알고리즘 001 | k번째 약수

1. 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

<내 코드>

import sys
sys.stdin=open("input.txt", "rt")

a, b=map(int, input().split())

lst = []
for i in range(1,a+1) :
    if a%i==0 :
        lst.append(i)
if len(lst)<b :
    print(-1)
else : 
    print(lst[b-1])

문제에서 변수는 n,k 라고 지정했었다 - 문제를 잘 읽자
n, k = map(int, input().split())

<풀이>

import sys
sys.stdin = open("input.txt", "rt")

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

<반성점>

• 리스트에 약수를 넣게 되면 모집단이 작을 때는 괜찮지만 그 숫자가 커지면 탐색하는 시간이 길어지기 때문에 비효율적
• 필요한 것은 k번째까지이기 때문에 그 이후는 break로 끝내는 것이 좋으나 사용하지 않음

<배운점>

• a, b = map(int, input().split())
• for ~ else: 반복문에서 break가 실행되지 않았을 때의 동작을 else에서 정의한다. (else는 for와 같은 indentation

<2회독 내 풀이>

n, m = map(int, input().split())
a=sorted(list([i+1 for i in range(n) if n%(i+1)==0]))
if m > len(a) :
    print(-1)
else : 
    print(a[m-1])

<추가 배운 것>

• list( [ i * 2 if i>3 else i for i in temp ] )
  : 'i에 2를 곱해서 리스틀 만들건데, 3보다 크면 하고 아니면 그대로 넣어. i는 for문으로 temp에서 가져와'

• list( [ i for i in temp if i > 3 ])
  : else 없을 땐 if를 뒤에다 쓴다

링크텍스트