just inside

[백준] Bronze 2. 2231 - 분해합 python 본문

coding test/완전탐색

[백준] Bronze 2. 2231 - 분해합 python

방울도마도 2024. 7. 12. 09:39
728x90

문제 링크

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을 출력한다.


제출 코드

import sys

input = sys.stdin.readline

n = int(input().rstrip())
found = False

for i in range(1, n):
    n_sum = i + sum(map(int, str(i)))
    if n_sum == n:
        print(i)
        found = True
        break

if not found:
    print(0)

 

풀이

  • 1부터 n-1까지의 숫자들에 대해 생성자 여부를 확인한다. - 브루트포스 알고리즘 (완전 탐색)
  • 생성자를 찾지 못하는 경우를 처리하기 위해 found 변수 사용한다.
  • i 값과 각 자릿수의 합이 입력받은 n과 같다면, 생성자를 발견한 것이므로 i를 출력하고 found=True 처리 해준다.

 

알아둘 사항

  • map(int, str(i)) 하면 각 자릿수를 숫자로 한번에 처리 가능. 굳이 list로 바꾸어주지 않아도 sum 연산 가능.

틀린 코드

  • while 문 사용시 - 시간초과
import sys

input = sys.stdin.readline

n = int(input().rstrip())
i = 1

while True:
    split_sum = i
    for num in str(i):
        split_sum += int(num)
    if split_sum == n:
        print(i)
        break
    i += 1
    if n == i:
        print(0)
        break

 

728x90