给定长度为n(1e5)的数组,必须选择k(20)个不同位置的元素,使其增加x,其他位置的元素都减少x,求进行操作后数组的最大连续区间和

本文介绍了一种使用C++编程语言解决的一个整数问题,通过动态规划方法计算在给定数组中,对每个位置进行加x或减x操作后,使得前k个数字中有特定数量在所选区间内的最大可能值。

题目

思路:

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pb push_back
#define fi first
#define se second
#define lson p << 1
#define rson p << 1 | 1
const int maxn = 1e6 + 5, inf = 1e18 + 5, maxm = 4e4 + 5, mod = 998244353, N = 1e6;
int a[maxn], b[maxn];
bool vis[maxn];
int n, m;
string s;
int f[maxn][25][3];//f[i][j][t]表示前i个数字中,已经用了j次,当前位置的状态是t(0:当前位置在所选区间之前,1:在所选区间中,2:在所选区间后)

void solve(){
    int res = 0;
    int k;
    int x;
    int q;
    cin >> n >> k >> x;
    for(int i = 1; i <= n; i++){
        cin >> a[i];
    }
    for(int i = 0; i <= n; i++){
        for(int j = 0; j <= k; j++){
            for(int t = 0; t < 3; t++){
                f[i][j][t] = -inf;//无效状态赋值为无穷小,防止影响状态转移
            }
        }
    }
    f[0][0][0] = 0;
    for(int i = 1; i <= n; i++){
        for(int j = 0; j <= k; j++){
            for(int t = 0; t < 3; t++){
                for(int jj = j; jj >= max(0LL, j - 1); jj--){
                    int val = (jj == j ? (a[i] - x) : (a[i] + x));
                    for(int tt = 0; tt <= t; tt++){
                        f[i][j][t] = max(f[i][j][t], f[i - 1][jj][tt] + (t == 1 ? val : 0));
                    }
                }
                // cout << i << ' ' << j << ' ' << t << ' ' << f[i][j][t] << '\n';
            }
        }
    }
    cout << max(f[n][k][1], f[n][k][2]) << '\n';
}
    
signed main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    // fac[0] = 1;
    // for(int i = 1; i <= N; i++){
    //     fac[i] = fac[i - 1] * i % mod;
    // }
    // inv[N] = qpow(fac[N], mod - 2);
    // for(int i = N - 1; i >= 0; i--){
    //     inv[i] = inv[i + 1] * (i + 1) % mod;
    // }
    int T = 1;
    cin >> T;
    while (T--)
    {
        solve();
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

__night_

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值