数字和为sum的方法数（DP+01背包方案计数）

时间限制：1秒  空间限制：32768K  热度指数：11166

 算法知识视频讲解

题目描述

给定一个有n个正整数的数组A和一个整数sum,求选择数组A中部分数字和为sum的方案数。
当两种选取方案有一个数字的下标不一样,我们就认为是不同的组成方案。

输入描述:

输入为两行:
 第一行为两个正整数n(1 ≤ n ≤ 1000)，sum(1 ≤ sum ≤ 1000)
 第二行为n个正整数A[i](32位整数)，以空格隔开。

输出描述:

输出所求的方案数

示例1

输入

5 15 5 5 10 2 3

输出

4

【分析】DP，类似背包问题

法一：朴素解法  dp[i][j]-前i个数组成和为j的方案数

#include <stdio.h>
#define maxn 1005
typedef long long ll;
ll n,sum;
ll a[maxn],dp[maxn][maxn];  //dp[i][j]-前i个数组成和为j的方案数 
int main()
{
	int i,j;
	scanf("%lld %lld",&n,&sum);
	for(i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	//初始化:前i个数组成和为0的方案数为1(什么数都不取) 
	for(i=0;i<=n;i++)
		dp[i][0]=1;
	//初始化:前0个数组成和为j(j!=0)的方案数为0(不可能凭空存在) 
	for(j=1;j<=sum;j++)
		dp[0][j]=0;
	//考虑是否使用第i个数，分两种情况 
	for(i=1;i<=n;i++)
	{
		for(j=0;j<=sum;j++)
		{
			//可以使用第i个数，则方案数为使用第i个数+不使用第i个数组成和为j的方案数
			//也就是说，用前i-1个数组成j的方案数+用前i-1个数组成j-p[i]的方案数 
			if(j>=a[i])
				dp[i][j]=dp[i-1][j]+dp[i-1][j-a[i]];
			//不可以使用第i个数 
			else
				dp[i][j]=dp[i-1][j];
		}
	}
	printf("%lld\n",dp[n][sum]);
	return 0;
}

法二——优化解法（优化空间复杂度）  dp[i]-前n个数组成和为i的方案数

#include <stdio.h>
#define maxn 1005
typedef long long ll;
ll n,sum;
ll a[maxn],dp[maxn];  //dp[i]-前n个数组成和为i的方案数 
int main()
{
	int i,j;
	scanf("%lld %lld",&n,&sum);
	for(i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	//初始化:组成和为0的方案数为1(n个数什么都不取) 
	dp[0]=1;
	for(i=1;i<=n;i++)
		for(j=sum;j>=a[i];j--)
			dp[j]=dp[j-a[i]]+dp[j];
	printf("%lld\n",dp[sum]);
	return 0;
}