[백준] 17626번 파이썬

문제: https://www.acmicpc.net/problem/17626'

문제 해결 방법

1. 문제 풀이를 이해하는데 매우 오랜 시간이 걸렸다. 구글링을 통해 여러 사람들의 내용을 살펴봤으나, 이해가 잘 되지 않았다. 다행히 몇 시간 정도 다시 생각해보니 이해가 되어 해결 할 수 있었다.
2. dp가 N+1개의 길이인 것은 자연수가 1부터 시작해서 해당 숫자의 경우의 수에 맞게끔 접근하기 위해서다.
3. 자연수 1은 1의 제곱으로만 나타낼 수 있다.
4. 주어진 N의 dp는 dp[N]이기에 첫번째 반복문은 N + 1까지 동작한다.
5. 이제 주어진 N의 제곱근 만큼의 범위까지 탐색하면서 각 N의 제곱근보다 작은 자연수들의 최소 거듭제곱 경우의 수를 탐색한다. 예로 i가 2라면, j는 1로 dp[1]이 되어 dp[2] = 2가 된다.
6. 만약 주어진 N이 어떤 자연수의 제곱이라면 dp[i - (j ** 2)] 는 dp[0]이기 때문에 마지막 dp[i] = minimum + 1의 결과로 값이 1이 된다.

import sys
input = sys.stdin.readline

N = int(input())
dp = [0] * (N + 1)  # 자연수 1부터 시작, 인덱스 맞춰주기
dp[1] = 1

for i in range(2, N + 1):
    minimum = sys.maxsize

    # 1부터 탐색
    for j in range(1, int(i ** 0.5) + 1):
        minimum = min(minimum, dp[i - (j ** 2)])
    dp[i] = minimum + 1

print(dp[N])