백준 1699번 제곱수의 합 C++

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

1699번: 제곱수의 합

 

dp[n]의 최솟값은 dp[n - a^2] + 1 중 최솟값이다. (a^2 <=n) 

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    int n;
    cin >> n;

    vector<int> dp(n + 1);
    for (int i = 1; i <= n; i++) dp[i] = i;
    for (int i = 1; i <= n; i++) {
        int a = 1;
        while (true) {
            int b = pow(a, 2);
            if (b == i) {
                dp[i] = 1;
                break;
            }
            if (b > i) break;
            dp[i] = min(dp[i], dp[i - b] + 1);
            a++;
        }
    }
    cout << dp[n];
}