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

本文最后更新于:3 years ago

题目链接:点击跳转

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

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

代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#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;
}