题目链接
https://www.luogu.com.cn/problem/P7335
思路
我们很容易想到 d p i , j dp_{i,j} dpi,j表示前 i i i个数选取 j j j个区间的最大值,答案为 d p n , k dp_{n,k} dpn,k。
但是,每一次转移都是一段连续的区间,通过枚举点 k k k使得 k < i k<i k<i,则状态转移方程为:
d p i , j = d p k , j − 1 + f k + 1 , j dp_{i,j} = dp_{k,j-1}+f_{k+1,j} dpi,j=dpk,j−1+fk+1,j,其中 f l , r f_{l,r} fl,r表示区间 [ l , r ] [l,r] [l,r]内一段异或区间的异或和的最大值。
我们可以使用前缀异或和,这样我们就可以把区间转为两个数的异或和,因此 f f f数组可以直接预处理得到。
因为数据随机,我们通过打表发现序列 f f f有很长一段的值是连续且相等的,又因为 d p dp dp一维数组单调不降,我们可以合并这些连续段。
时间复杂度: O ( n 2 l o g n ) O(n^2logn) O(n2logn)
代码
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 3e3 + 5;
int n, k;
int a[N], sum[N], f[N][N], dp[N][2];
vector<pair<int, int>>g[N];
void solve()
{
cin >> n >> k;
for (int i = 1; i <= n; i++)
{
cin >> a[i];
sum[i] = sum[i - 1] ^ a[i];
}
for (int i = 1; i <= n; i++)
{
int res = 0;
for (int j = i - 1; j >= 0; j--)
{
res = max(res, sum[i] ^ sum[j]);
f[j + 1][i] = max(res, f[j + 1][i - 1]);
}
}
for (int i = 1; i <= n; i++)
{
int r = i;
for (int j = i - 1; j >= 1; j--)
{
if (f[j][i] != f[j + 1][i])
{
g[i].push_back({j + 1, r});
r = j;
}
}
g[i].push_back({1, r});
}
int now = 1;
for (int i = 1; i <= k; i++)
{
for (int j = 1; j <= n; j++)
{
for (auto val : g[j])
{
dp[j][now] = max(dp[j][now], dp[val.second - 1][now ^ 1] + f[val.second][j]);
}
}
now ^= 1;
}
cout << dp[n][now ^ 1] << endl;
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0), cout.tie(0);
int test = 1;
// cin >> test;
for (int i = 1; i <= test; i++)
{
solve();
}
return 0;
}