数字组合
给定 N 个正整数 A1,A2,…,AN,从中选出若干个数,使它们的和为 M,求有多少种选择方案。
输入格式
第一行包含两个整数 N 和 M。
第二行包含 N 个整数,表示 A1,A2,…,AN。
输出格式
包含一个整数,表示可选方案数。
输入样例:
4 4
1 1 2 2
输出样例:
3
题意分析:可以把数字看成体积,题目也即是要求恰好装满的情况有多少种。
因为要求的是恰好装满的情况,所以初始化的时候自然不能初始化f[0] = 1以外的情况,也就是说只能从0开始转移,自然最后要求的就是f[m]的值,此时背包所用体积一定是m。
状态转移,f[i][j] = f[i-1][j]+f[i-1][j-v[i]]
但是可以压成一维:原因自然是每次更新的时候都只会用到上一行数据,因此只存一行数据就可。但是需要满足01背包的特点——从后往前更新,这样确保更新时用到的都是上一行的数据。
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
using namespace std;
const int N = 10010;
int n,m; //一共是n个数,和为m
int num[N];
int f[N];
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>num[i];
f[0] = 1;
for(int i=1;i<=n;i++)
for(int j=m;j>=num[i];j--)
f[j] += f[j-num[i]];
cout<<f[m];
return 0;
}