C. Increase Subarray Sums - 思维

题面

分析

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

代码
#include 

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

你可能感兴趣的:(算法,c++,思维)