[백준] 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을 출력한다.

---

def solution():
    n = int(input())
    for num in range(1, n):
        nums_array = [num]
        nums_array += list(map(int, list(str(num))))
        if sum(nums_array) == n:
            print(num)
            return

    print(0)


if __name__ == "__main__":
    solution()

해설

• 자연수 n을 입력 받은 뒤 1 부터 (n - 1)까지 순회하며 아래 작업을 수행한다.
• 순회하는 수 num과 num의 각 자리수를 담을 nums_array 리스트를 생성하여 해당 리스트의 합이 n과 같은 경우 num을 출력하고 함수를 빠져나가도록 한다.
• 만약 빠져나가지 못하고 순회가 끝난 경우 마지막에 0을 출력하도록 한다.