lc279.完全平方数

题目描述

给定正整数 n，找到若干个完全平方数（比如 1, 4, 9, 16, …）使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。

示例1

输入: n = 12
输出: 3
解释: 12 = 4 + 4 + 4.

示例2

输入: n = 13
输出: 2
解释: 13 = 4 + 9.

方法一：暴力求解

numSquares(n)=min(numSquares(n-k) + 1) K是其中<=n的一个完全平方数；遍历每一个k，找出min(numSquares(n-k) + 1

#include"iostream"
#include"vector"
#include"cmath"
#include"algorithm"
using namespace std;
vector<int>nums;
int minsquares(int k)
{   
	int j;int i;
	for (i = 0;i < nums.size();i++)//如果k是完全平方数，返回值应该为1
	{
		if (nums[i] == k)
		{
			return 1;
		}
	}
	int minnum = INT_MAX;
	for (j = 0;j < nums.size();j++)
	{
		if (k < nums[j])
			break;
		int newmin = minsquares(k - nums[j]) + 1;//递归地计算k - nums[j]的平方数个数
		minnum = min(newmin, minnum);
	}
	return minnum;
}
int main()
{
	int n;
	cin >> n;
	int i = 1;
	while (i * i < n)
	{
		nums.push_back(i * i);
		i++;
	}
	cout << minsquares(n) << endl;
}

方法二 动态规划法
保存计算过程中的中间解，体现在dp数组上；为了表示numSquares(n-k) + 1(当k==n时，也就是n本身就是个完全平方数的情况)，应该初始化dp[0]=0;之后与暴力求解步骤几乎相同。

int main()
{
	int n;int j;
	cin >> n;
	vector<int>nums;
	vector<int>dp(n+1,INT_MAX);//初始化dp数组
	dp[0]=0;
	int i = 1;
	while (i * i < n)
	{//得到完全平方数列表
		nums.push_back(i * i);
		i++;
	}
	for(i=1;i<n+1;i++)
	{
		for (j = 0;j < nums.size();j++)
		{
			if (i< nums[j])
				break;
			dp[i] = min(dp[i],dp[i - nums[j]]+ 1);//动态规划递推式
		}
	}
	cout << dp[n];
	return 0;
}

暴力递归+贪心策略

在暴力递归时我们是从minsquares(n)->minsquares(n-1)->minsquares(n-2)…minsquares(1)
这个方法我们直接从平方数的个数入手，让divide(n,count)表示 数n,由count个平方数组成;divide(n-k,count-1),表示数n,可以由原来的n,分为n-k和k,k和n-k的平方数个数最少为也就是count-1.
让个数从1…n, 直到找到一个count,使它符合每一项都是平方数。我们就可以直接跳出循环表示找到了。此时找到的就是最小个数。具体如下图所示：注意体会count值的变化

#include"iostream"
#include"vector"
using namespace std;
vector<int>nums;
bool exist(int n)//判断某个数是不是完全平方数
{
	int j;
	for (j = 0;j < nums.size();j++)
	{
		if (nums[j] == n)
			return true;
	}
	return false;
}
bool divided(int n, int count)
{
	int i;int k;
	if (count == 1)//递归边界，如果只由一个数字组成，看数本身是不是完全平方数
		return exist(n);
	for (i = 0;i < nums.size();i++)
	{
		if (divided(n - nums[i], count -1))
			return true;
	}
	return false;
}
int main()
{
	int n;
	cin >> n;
	int i = 1;
	while (i * i < n)
	{
		nums.push_back(i * i);
		i++;
	}
	int count;
	for (count = 1;count <= n;count++)
	{
		if (divided(n, count))
		{ 
			cout << count;
		}
	}
	return 0;
}