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

最新推荐文章于 2022-04-11 19:52:20 发布

i.ajls

最新推荐文章于 2022-04-11 19:52:20 发布

阅读量1k

点赞数

分类专栏：补题文章标签：算法 c语言动态规划

本文链接：https://blog.csdn.net/weixin_43823753/article/details/122746478

版权

补题专栏收录该内容

35 篇文章 0 订阅

订阅专栏

题目链接:点击跳转

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

思路：:从前往后处理，如果当前这个重量增加了西瓜，因为可以买半个，那么一定会影响到前面的值导致错误，所以每个重量都认为从半个西瓜得到的，每增加一个使用上一题的dp(链接:点击跳转)来记录当前西瓜所能组成的数量，然后和输入数据比较，如果不够，就继续添加。

代码如下：

#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;
}