[Algorithm] 백준 2231번 분해합 (Python)

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

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

216

예제 출력 1

198

풀이

설명

• i_sum에 i의 각 자릿수의 합을 저장한다.
• total에 i와 i의 각 자릿수의 합을 더해 저장한다.
• total과 처음 입력한 수가 동일하면 현재 i값을 출력한다.
• i와 처음 입력한 수가 동일하면 생성자가 없다는 의미이므로 0을 출력한다.

코드

N = int(input())

for i in range(1, N+1):
    i_sum = sum(map(int, str(i)))
    total = i_sum + i

    if total == N:
        print(i)
        break
    elif i == N:
        print(0)