TYVJ 1096 数字组合【0-1背包模型】

题目描述

在N个数中找出其和为M的若干个数。先读入正整数N（1<N<100）和M(1<M<10000)， 再读入N个正数（可以有相同的数字，每个数字均在1000以内）， 在这N个数中找出若干个数， 使它们的和是M， 把满足条件的数字组合都找出来以统计组合的个数，输出组合的个数（不考虑组合是否相同）。要求你的程序运行时间不超过1秒。

输入格式

第一行是两个数字，表示N和M。
第二行起是N个数。

输出格式

就一个数字，表示和为M的组合的个数。

样例数据

输入样例 #1	输出样例 #1
4 4 1 1 2 2	3

 

题解：n 个正整数看作 n 个物品，m 看作背包容量。设 dp[ j ] 表示和为 j 的方案数。dp[0] = 1，即一个也不选的方案数为 1。

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 1000+7;
int dp[maxn], a[maxn];
int main()
{
    int n, m;
    cin >> n>> m;
    for(int i = 1; i <= n; i++)
        cin >> a[i];
    dp[0] = 1;
    for(int i = 1; i <= n; i++){
        for(int j = m; j >= a[i]; j--)
            dp[j] += dp[j-a[i]];
    }
    cout << dp[m] <<endl;
    return 0;
}