2022牛客寒假算法基础集训营3 C 智乃买瓜(another version)

本文最后更新于:3 years ago

题目链接:点击跳转

题意::和上题的区别就是现在是已知每个重量的西瓜能有多少种取法,求该组西瓜的重量(答案不唯一)

思路::从前往后处理,如果当前这个重量增加了西瓜,因为可以买半个,那么一定会影响到前面的值导致错误,所以每个重量都认为从半个西瓜得到的,每增加一个使用上一题的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
37
38
39
40
41
42
43
44
45
46
47
#include<bits/stdc++.h>

using namespace std;
typedef long long ll;

const ll MOD = 1e9 + 7;
const int N = 1e5 + 5e2;
int num[N];

signed main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
int m;
cin >> m;
for (int i = 1; i <= m; i++) {
cin >> num[i];
}
vector<int>v(N, 0), res;
v[0] = 1;
for (int i = 1; i <= m; i++) {
while (num[i] > v[i]) {
int a = i << 1;
res.push_back(a);
vector<int>p = v, q = v;
p[0] = q[0] = 1;
for (int j = 0; j <= m; j++) {
p[j + a] = (v[j + a] + v[j]) % MOD;
}
for (int j = 0; j <= m; j++) {
q[j + a / 2] = (v[j + a / 2] + v[j]) % MOD;
}
for (int j = 1; j <= m; j++) {
v[j] = (p[j] + q[j] - v[j] + MOD) % MOD;
}
}
}
cout << res.size() << endl;
bool f = false;
for(auto i : res) {
if (f) cout << ' ';
f = true;
cout << i;
}
cout << endl;
return 0;
}

本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!