[Python] 백준 2231번 분해합

문제

어떤 자연수 N이 있을 때, 그 자연수 N의 분해합은 N과 N을 이루는 각 자릿수의 합을 의미한다. 

어떤 자연수 M의 분해합이 N인 경우, M을 N의 생성자라 한다. 

예를 들어, 245의 분해합은 256(=245+2+4+5)이 된다. 따라서 245는 256의 생성자가 된다. 

물론, 어떤 자연수의 경우에는 생성자가 없을 수도 있다. 반대로, 생성자가 여러 개인 자연수도 있을 수 있다.
자연수 N이 주어졌을 때, N의 가장 작은 생성자를 구해내는 프로그램을 작성하시오.

 

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

2231번: 분해합

입력

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

출력

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

예제

__input__
216
__output__
198

문제 풀이

N = int(input())

for i in range(1,N+1):
    tmp = sum(map(int,str(i)))  # N의 자릿수 총합(iterable한 map 객체의 sum으로 구함)
    ans = i + tmp
    if ans == N:
        print(i)
        break
    if i == N:					# i가 N에 도달할 때까지 생성자 조건에 부합하지 못할 경우
        print(0)

다만, 시간이 너무 오래 소요되어 다른 분들의 제출 결과를 추가로 확인해보았습니다.

N = int(input())
min_N = abs(N - len(str(N)) * 9)

for i in range(min_N, N+1):
	tmp = sum(map(int, str(i)))
    ans = i + tmp
    if ans == N:
    	print(i)
        break
	if i == N:
    	print(0)

분해합을 구하기 위해 1부터 생성자 + 생성자 자릿수의 합을 구하는데요.

이 과정을 단축하기 위해 각 자릿수별로 나올 수 있는 최대수인 9를 빼주면 보다 좁은 범위를 탐색할 수 있습니다.

다만, 음수가 될 수 있기에 abs 함수를 통해 절댓값으로 만들어주었음을 볼 수 있습니다.

ex) N = 15, min_N = |-3|

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,

혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.