2022牛客寒假算法基础集训营3 B 智乃买瓜

题目链接:点击跳转

题意：有n个西瓜，每个可以买整个和买半个，问如果他想要购买西瓜的重量和分别为k=1,2,3…M时，有多少种购买西瓜的方案。

思路：将每个西瓜和已经买下的使用dp进行组合，注意半个和整个要独立存入

代码如下：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const ll MOD = 1e9 + 7;
const int N = 1e3 + 5e2;

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);cout.tie(nullptr);
    int n, m;
    cin >> n >> m;
    vector<int>res(N << 1, 0); //多开一些后面不需要判断越界
    for (int i = 0; i < n; i++) {
        int a;
        cin >> a;
        res[0] = 1;
        vector<int>v = res, q = res;
        v[0] = q[0] = 1; //初始化为1保证存入的本身重量加1个
        for (int j = 0; j <= m; j++) {
            v[j + a] = (res[j + a] + res[j]) % MOD; //记录每个组合后的数量
        }
        for (int j = 0; j <= m; j++) {
            q[j + a / 2] = (res[j + a / 2] + res[j]) % MOD;
        }
        for (int j = 1; j <= m; j++) {
            res[j] = (v[j] + q[j] - res[j] + MOD) % MOD; //加MOD防止变负数
        }
    }
    for (int i = 1; i <= m; i++) {
        if (i != 1) cout << ' ';
        cout << res[i];
    }
    cout << endl;
    return 0;
}