[백준/python] 2231 분해합: 자릿수 덧셈 (sum, map 함수 사용)

문제

---

어떤 자연수 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을 출력한다.

 

 

 

N=int(input())
N2=1
result=0
digitList=[]

while(1):
  N2+=1
  digitList=list(map(int, str(N2))) #
  result+=N2
  result+=sum(digitList)
  if result==N:
    print(N2)
    break
  if N2==N:
    print(0)
    break
  result=0

 

N2 + N2의 각 자릿수를 했을때 N과 같으면 그것이 생성자. (N2는 N의 생성자)

라는 아이디어를 갖고 코드를 짰다.

 

하지만 이렇게 코드를 제출하니 시간초과가 떴다. (ㅠㅠ)

 

 

 

N=int(input())
result=0

for N2 in range(1,N+1):
  digitSum=sum(map(int, str(N2))) #자릿수를 더함
  result=N2+digitSum
  if result==N:
    print(N2)
    break
  if N2==N:
    print(0)
    break

 

이렇게 하니 맞았다.

확실히 변수 선언이나 불필요한 함수 사용이 줄었다.