思路:

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

被折叠的 条评论
为什么被折叠?



