本文最后更新于: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; }
|