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

입력

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

출력

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

예제 입출력

풀이 1 (정답)

n, k = map(int, input().split())
factors = [] # 약수 배열
for i in range(1, n+1): # 약수 구하기
	if n % i == 0:
		factors.append(i)

if len(factors) < k: print(0)
else: print(factors[k-1])

index 범위 때문에 한 번 틀렸다.
정수 n이 주어졌을 때, 정수 n을 1부터 n까지 나눈 몫을 구해야 하므로
범위를 range(1, n+1) 으로 설정해주어야 한다.

풀이 2 (정답, list comprehension)

n, k = map(int, input().split())
factors = [i for i in range(1, n+1) if n % i == 0]

if len(factors) < k: print(0)
else: print(factors[k-1])

풀이 1에서 약수를 구하는 부분(3~5행)을 list comprehension을 이용해 한 줄로 줄였다.

풀이 3 (정답, try-except)

n, k = map(int, input().split())
factors = [i for i in range(1, n+1) if n % i == 0]
try: print(factors[k-1])
except: print(0)

풀이 2의 if-else문을 try-except문으로 대체하였다.
문제를 처음 볼 때 얼핏 떠올렸는데,
굳이 싶어서 안썼다가 숏코딩 첫 페이지에 있길래 작성해보았다.

오늘의 교훈

• try-except문을 자주 써보자.
• list comprehension에서 for문과 if문을 동시에 사용할 수 있다.

[x for x in 범위 if 조건]
# e.g.
result = [x for x in range(1, 10) if 10 % x == 0]