CH5201 数组组合01背包

Posted 2021-01-11 wyboooo

5201 数字组合 0x50「动态规划」例题

描述

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

输入格式

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

输出格式

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

样例输入

4 4
1 1 2 2

样例输出

3

 

思路：

非常基础的01背包问题 dp[i][j]为前i个数选出和为j的若干个数的方案数

实际上i的这一维是不需要的，因为i+1都是在i的基础上做的

采用倒序循环，循环到j时，后半部分j~m处于第i阶段，前半部分处于第i-1阶段

 1 #include <bits/stdc++.h>
 2 #include<iostream>
 3 #include<cmath>
 4 #include<algorithm>
 5 #include<stdio.h>
 6 #include<cstring>
 7 #include<map>
 8 
 9 #define inf 0x3f3f3f3f
10 using namespace std;
11 typedef long long LL;
12 
13 int n, m;
14 const int maxn = 105;
15 int v[maxn], dp[10005];
16 
17 int main()
18 {
19     scanf("%d%d", &n, &m);
20     for(int i = 0; i < n; i++){
21         scanf("%d", &v[i]);
22     }
23 
24     memset(dp, 0, sizeof(dp));
25     dp[0] = 1;
26     for(int i = 0; i < n; i++){
27         for(int j = m; j >= v[i]; j--){
28             dp[j] += dp[j - v[i]];
29         }
30     }
31     printf("%d
", dp[m]);
32     return 0;
33 }