278. 数字组合 - AcWing题库 给定 N 个正整数 A1,A2,…,AN,从中选出若干个数,使它们的和为 M,求有多少种选择方案。

输入格式

第一行包含两个整数 N 和 M。

第二行包含 N 个整数,表示 A1,A2,…,AN。

输出格式

包含一个整数,表示可选方案数。

数据范围

1≤N≤100,
1≤M≤10000,
1≤Ai≤1000,
答案保证在 int 范围内。

输入样例:

4 4
1 1 2 2

输出样例:

3

思路

数字看做体积, 求体积恰好为M的方案数。

状态表示:f[i][j] 前i个数, 总和为j的方案数 状态计算:

  • 不包含a[i]且总和为j, f[i - 1][j]
  • 包含a[i]且总和为j, f[i - 1][j - a[i]] 方案数就是两个方案加起来。

初始化: f[i][0] 表示从前i个选总和为j的方案数, 为 1 f[0][j] 表示选0个总和为j的方案数, 为 0

用到的都是上一层的值, 可以进行一维优化。

代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
 
const int N = 1e4 + 10;
int f[N];
int n,m;
 
int main()
{
    cin >> n >> m;
    f[0] = 1;
    for(int i = 1; i <= n; i++)
    {
        int t;
        cin >> t;
        for(int j = m; j >= t; j--)
            f[j] = f[j] + f[j-t];
    }
    cout << f[m] << endl;
    return 0;
}