题目描述
给定正整数 n,找到若干个完全平方数(比如 1, 4, 9, 16, ...
)使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。
示例 1:
输入: n = 12 输出: 3 解释: 12 = 4 + 4 + 4.
示例 2:
输入: n = 13 输出: 2 解释: 13 = 4 + 9.
解题思路
本人不才,看不懂大佬的四平定理,但是本题可以使用类似完全背包的思想解决。定义dp[i]表示数i需要的最小平方数,那么状态转换方程为dp[i] = min(dp[i],dp[i-j*j]+1)。
例如i = 15,
dp[10] = min(dp[10],dp[10-1*1]+1)
dp[10] = min(dp[10,dp[10-2*2]+1])
dp[10] = min(dp[10],dp[10-3*3]+1)
int numSquares(int n) {
int dp[n+1] = {0};
fill(dp,dp+n+1,INT_MAX);
dp[0] = 0;dp[1] = 1;
for(int i=2;i<=n;i++){
for(int j=1;j*j<=i;j++){
dp[i] = min(dp[i],dp[i-j*j]+1);
}
}
return dp[n];
}