C. Increase Subarray Sums - 思维

.y.a.o.

于 2023-11-26 15:28:28 发布

阅读量437

点赞数 10

文章标签：算法 c++ 思维

本文链接：https://blog.csdn.net/m0_75129175/article/details/134628419

版权

文章介绍了如何使用前缀和和动态规划的方法解决一个题目，涉及到在给定数组中找到至少k个数，使得它们加上x后，数组的最大距离可以通过计算前缀和并维护最小前缀和来求得。给出了C++代码实现。

摘要由CSDN通过智能技术生成

题面

分析

最优方法就是至少选k个数，这k个数都可以加上x，一段的最大距离可以通过前缀和来实现，还需要一个数组记录当前最小的前缀和，每次只需要当前前缀和减去长度为k之前的最小前缀和即可。

代码

#include <bits/stdc++.h>

#define int long long

using namespace std;

void solve() {
    int n, x;
    cin >> n >> x;
    vector<int> a(n + 1);
    vector<int> g(n + 1);
    for(int i = 1; i <= n; i ++) {
        cin >> a[i];
        a[i] += a[i - 1];
        g[i] = min(g[i - 1], a[i]);
    }
    int ans = 0;
    for(int i = 0; i <= n; i ++) {
        for(int j = i; j <= n; j ++) {
            ans = max(ans, a[j] - g[j - i] + i * x);
        }
        cout << ans << ' ';
    }
    cout << "\n";
}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int T;
    cin >> T;
    while(T --) {
        solve();
    }
}