2231: 분해합 - python

https://www.acmicpc.net/problem/2231

문제

어떤 자연수 N이 있을 때, 그 자연수 N의 분해합은 N과 N을 이루는 각 자리수의 합을 의미한다. 어떤 자연수 M의 분해합이 N인 경우, M을 N의 생성자라 한다. 예를 들어, 245의 분해합은 256(=245+2+4+5)이 된다. 따라서 245는 256의 생성자가 된다. 물론, 어떤 자연수의 경우에는 생성자가 없을 수도 있다. 반대로, 생성자가 여러 개인 자연수도 있을 수 있다.

자연수 N이 주어졌을 때, N의 가장 작은 생성자를 구해내는 프로그램을 작성하시오.

입력

첫째 줄에 자연수 N(1 ≤ N ≤ 1,000,000)이 주어진다.

출력

첫째 줄에 답을 출력한다. 생성자가 없는 경우에는 0을 출력한다.

예제 입출력

풀이 1 (정답)

N = int(input())
ans = 0

for i in range(N):
	res = i
	for j in str(i):
		res += int(j)
	if res == N:
		ans = i
		break
print(ans)

어렵지 않은 문제였다.
i의 각 자리수를 더하기 위해 i를 문자열로 변환한 뒤 for문을 사용했다.
다만 for문을 두 번 사용해서인지 시간이 오래 걸렸다.

풀이 2 (정답, 참고한 풀이)

N = int(input())
res = 0
ans = 0

for i in range(N):
	i_list = list(map(int, str(i)))
	res = i + sum(i_list)
	if res == N:
		ans = i
		break
print(ans)

실행 속도를 개선하고자 for문을 하나 줄이고 싶었는데,
for문 없이 어떻게 각 자리수를 쪼개야 할지 몰랐다.

이때 minblog님의 코드에서 좋은 방법을 찾았다.
map 함수를 사용하면 int형을 유지한 채 list로 변환할 수 있었다.

결과적으로 풀이 1에 비해 244ms가 감소하였다.

오늘의 교훈

• map 함수를 사용하면 원하는 type의 list를 만들 수 있다.