首页 > 代码库 > 待字闺中之正数数组内和为指定数字的总数

待字闺中之正数数组内和为指定数字的总数

求正数数组内和为指定数字的合并总数

比如[5, 5, 10, 2, 3] 合并值为 15 : 有4种 : (5 + 10, 5 + 10, 5 + 5 + 2 + 3, 10 + 2 + 3)

分析:有的时候,一个题目不能够立刻想到比较优化的想法,就可以先找到一个解决方案,然后根据方案的不足进行优化。而且这个时候,逆转一下思路,便会柳暗花明。由递归到动态规划,不就是如此么?

我们设定f(index,sum)表示数组从index开始到结束组成的sum的总数。那么,f(index, sum)可以表示为什么呢? 我们这个题目,就是要最终求得f(0, 15),从头开始分析,最终组成的和为15的可能组合中,可能包含第0个元素,也可能不包含, 原始数组为A:

  • 当包含第0个元素时,剩下的表示为f(1, 15-A[0])
  • 不包含第0个元素时,剩下的表示为f(1, 15)

则,f(0, 15) = f(1, 15) + f(1, 15 - A[0])。依次递归。递归的终止条件是什么呢?对于f(index,sum):

  • 当和小于等于0的时候,f(index,sum) = 0
  • 当和小于sum的时候, f(index, sum) = f(index + 1, num);
  • 当和等于sum的时候,f(index, sum) = 1 + f(index + 1, sum);

但是,上面的条件,并没有使用题目中,数组全是正数,也就是存在负数也可以。如果仅仅是正数,后两个改为:

  • 当和小于sum的时候, f(index, sum) = 0;
  • 当和等于sum的时候,f(index, sum) = 1;

有一个条件,我们没有使用,也意味着提升的空间。

可是,上面的方案,时间复杂度是指数级。怎么做一些改进呢?一般在对一个算法进行优化的时候,有哪些思路呢?尤其是这种时间很恐怖的?我想很多同学都有这个经验,就是空间换时间。

大家可以想象动态规划的思想,大家看如下的状态转移方程:

dp[n][m]=dp[n-1][m]+dp[n-1][m-num[n-1]]

dp[n][m]表示前n个元素组成和为m的情况数。初始化dp[0][0]=1,其他为0。写出状态转移方程,大家也就明白了,为何要求全是正数了吧,直白一些,数组的索引,怎么可能为负呢?在计算的过程中,将和的情况保存下来,用空间换时间,整个算法的时间复杂度为O(n*m),不再是指数级。

具体代码如下:

int totalCountRecusive(vector<int>& data,int index,int sum)
{
	int length = data.size();
	if(sum == 0) return 1;//找到一个结果
	if(index == length)return 0;
	return totalCountRecusive(data,index+1,sum-data[index]) + totalCountRecusive(data,index+1,sum);//递归
}
int totalCountRecusive(vector<int>& data,int sum)//递归方法
{
	return totalCountRecusive(data,0,sum);
}
int totalCountDp(vector<int>& data,int sum)//动态规划方法
{
	int length = data.size();
	if(length <= 0)return 0;
	vector<vector<int> > dp(length+1);
	int i,j;
	for(i = 0;i <= length;i++)
	{
		vector<int> tmp(sum+1,0);<span style="font-family: Verdana, Arial, Tahoma, Helvetica, Georgia, sans-serif, STXihei, 华文黑体, Hei, 'Hiragino Kaku Gothic Pro', SimSun; font-size: 12px;">//初始化</span>
		dp[i] = tmp;
	}
	for(i = 0;i <= length;i++)dp[i][0] = 1;//初始化
	for(i = length-1;i >= 0;i--)
	{
		for(j = sum;j > 0;j --)
		{
			dp[i][j] = dp[i+1][j];
			if(j - data[i] >= 0)dp[i][j] += dp[i+1][j-data[i]];//分开,防止下标为负数
		}
	}
	return dp[0][sum];
}