279. 完全平方数-情况不对就返回无穷大

问题

给你一个整数 n ，返回 和为 n 的完全平方数的最少数量 。

完全平方数 是一个整数，其值等于另一个整数的平方；换句话说，其值等于一个整数自乘的积。例如，1、4、9 和 16 都是完全平方数，而 3 和 11 不是。

示例 1：
输入：n = 12
输出：3
解释：12 = 4 + 4 + 4

分析

dfs比较简单好写。递推还是得再学习一下怎么想。

代码

class Solution {
public:
    vector<int> nums;
    int n_n = 0;
    int mem[101][10010];
    const int INF = 1e9;
    int dfs(int x, int left) {
        if (x >= n_n) {return INF;}
        if (left == 0) {return 0;}
        if (mem[x][left]!=-1) {return mem[x][left];}
        int s = 0;
        if (left < nums[x]) {
            s = dfs(x+1, left);
        } else {
            s = min(dfs(x+1, left), dfs(x, left-nums[x])+1);
        }
        mem[x][left] = s;
        return s;
    }
    int numSquares(int n) {
        memset(mem, -1, sizeof mem);
        for (int i = 1; i*i <= n; i++) {
            nums.push_back(i*i);
        }
        n_n = nums.size();
        int res = dfs(0, n);
        return res;
    }
};

new

对于不合理的情况，也就是leftX != 0，那么让min起作用，就要返回一个无穷大。

class Solution {
private:
    static const int N = 110;
    int nums[N];
    int mem[N][10010];
    int MAXN = 1e9+10;
    int res = 0;
    int dfs(int x, int leftX) {
        if (x >= N) {return MAXN;}
        if (mem[x][leftX]) {return mem[x][leftX];}
        if (leftX == 0) {return 0;}
        int s = 0;
        if (leftX < nums[x]) {
            s = dfs(x+1, leftX);
        } else {
            s = min(dfs(x+1, leftX), dfs(x, leftX-nums[x])+1);
        }
        mem[x][leftX] = s;
        return s;
    }
public:
    int numSquares(int n) {
        memset(mem, 0, sizeof mem);
        for (int i = 0; i < N; i++) {
            nums[i] = i*i;
        }
        res = dfs(1, n); // 完全平方数从1开始，不能传进去0，否则报错
        return res;
    }
};