题目如下:
/**
给定一个有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
输出例子1:
4
*/
思路如下:
方案一:搜索+记忆化(N太大)
方案二:DP + 滚动数组减少空间
代码如下:
#include<stdio.h>
#include<iostream>
#define MAX_N 1005
#define MAX_SUM 1005
typedef long long LL;
using namespace std;
int values[MAX_N];
LL dp[MAX_N][MAX_SUM];
int main(){
int N, Sum;
scanf("%d%d", &N, &Sum);
for(int i=1; i<=N; i++)
scanf("%d", values+i);
for(int i=0; i<=N; i++)
dp[i][0]=1;
for(int i=1; i<=N; i++){
for(int j=1; j<=Sum; j++){
dp[i][j]=dp[i-1][j];
if(j>=values[i])
dp[i][j]+=dp[i-1][j-values[i]];
}
}
printf("%lld", dp[N][Sum]);
return 0;
}
网友评论