知识点

动态规划

思路

对于一个数n,它一定能由若干个完全平方数组成,对于j来说,它会由j-i*i转移过来。

设dp[i]代表组成i的最小平方数个数,由于此题需要输出数量尽量少的平方数组合,那么,我们可以先预处理n的平方组成数,然后逆序从大到小地用平方数来组成n,

合法条件为对dp[n-j * j]==dp[n-1],且n>=j * j

代码c++

#include <climits>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param n int整型 
     * @return int整型vector
     */
    vector<int> numSquares(int n) {
        vector<int>dp(n+1,INT_MAX);
        dp[0]=0;
        for(int i=1;i<=n;i++)
        {
           for(int j=1;j*j<=i;j++)
           {
            dp[i]=min(dp[i],dp[i-j*j]+1);
           }
        }
        vector<int>ans;
        while(n>0)
        {
            for(int i=sqrt(n);i>=1;i--)
            {
                if(dp[n-i*i]+1==dp[n]&&n>=i*i)
                {
                    ans.push_back(i*i);
                    n-=i*i;
                    break;
                }
            }
        }
        sort(ans.begin(),ans.end());
        return ans;
    }
};