题目描述:
给定N个正整数A1,A2,…,AN,从中选出若干个数,使它们的和为M,求有多少种选择方案。
输入格式
第一行包含两个整数N和M。
第二行包含N个整数,表示A1,A2,…,AN。
输出格式
包含一个整数,表示可选方案数。
数据范围
1≤N≤100,
1≤M≤10000,
1≤Ai≤1000
输入样例:
4 4
1 1 2 2
输出样例:
3
分析:
本题与AcWing 900 整数划分问题类似,只不过整数划分问题是参考了完全背包问题,而本题每个数最多选一次,所以相当于01背包问题。由潜水员问题得出背包问题可衍生出三种问题,第一个是与常规背包问题类似,求体积不超过v的最大价值,第二个是求体积不小于v的最小价值,第三个是求体积恰好是v的最大最小价值或者方案数。
本题就属于第三种衍生问题,状态表示f[i][j]表示在前i个数中选择出总和恰好是j的若干个数的方案数。状态转移方程为f[i][j] = f[i-1][j] + f[i-1][j-v]。f[0][0]表示一个数都不选总和是0,所以方案数为1,f[0][j](j > 0)表示一个数都不选总和大于0,是不可能的状态,所以方案数为0.即f[0][0] = 1,f[0][j] = 0是本题的边界状态。使用滚动数组实现的代码如下:
#include <iostream>
using namespace std;
const int N = 105,M = 10005;
int v[N],f[M];
int main(){
int n,m;
cin>>n>>m;
for(int i = 1;i <= n;i++) cin>>v[i];
f[0] = 1;
for(int i = 1;i <= n;i++)
for(int j = m;j >= v[i];j--)
f[j] += f[j - v[i]];
cout<<f[m]<<endl;
return 0;
}
当然,我们也可以完全参考01背包问题,定义f[i][j]为前i个数中选出总和不超过j的方案数,状态转移方程同样是f[i][j] = f[i-1][j] + f[i-1][j-v]。边界状态为f[0][0]=1表示一个数都不选总和不超过0的方案数,同理f[0][j] = 1,(j > 0)。这是因为修改了状态表示的定义,一个数都不选,总和肯定不会超过正数,是合法状态。此时我们要求的是总和恰好是m的方案数ans,则ans = f[n][m] - f[n][m-1],因为选择的都是整数,总和不超过m的方案数减去总和不超过m - 1的方案数自然就是总和恰好是m的方案数了,这里的思想类似于前缀和。
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 105,M = 10005;
int v[N],f[M];
int main(){
int n,m;
cin>>n>>m;
for(int i = 1;i <= n;i++) cin>>v[i];
for(int i = 0;i <= m;i++) f[i] = 1;
for(int i = 1;i <= n;i++){
for(int j = m;j >= v[i];j--){
f[j] += f[j - v[i]];
}
}
cout<<f[m] - f[m-1]<<endl;
return 0;
}