Codeforces Round #416 (Div. 2) - C. Vladik and Memorable Trip 动态规划

题目链接:C. Vladik and Memorable Trip

题目大意

给你一个长度为n的数组a[i], 其中n<=5000, a[i]<=5000, 要求分出n个连续子序列, 要求如果这个子序列中出现了某个数字x, 那么这个子序列必须包括整个数组中的所有x, 不一定要把全部数组元素都用上, 将每个子序列包含的数字集合中每个数字异或起来为这个子序列的价值, 求最大总价值

思路

定义dp[i] := a[1]-a[i]能得到的最大价值
那么dp[i] = max{dp[i-1], //不使用a[i]
dp[j-1] + a[j] ^ a[j+1] ^ .. ^ a[i]//[j, i] 为合法区间
}

代码

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
const int MAXN = 6000, INF = 0X3F3F3F3F;
int n, l[MAXN], r[MAXN], a[MAXN], dp[MAXN];
bool vis[MAXN];

int main()
{
    scanf("%d", &n);
    memset(l, INF, sizeof(l));
    memset(r, 0, sizeof(r));
    for(int i=1; i<=n; ++i)
    {
        scanf("%d", a+i);
        l[a[i]] = min(l[a[i]], i);
        r[a[i]] = max(r[a[i]], i);
    }
    for(int i=1; i<=n; ++i)
    {
        dp[i] = max(dp[i], dp[i-1]);
        int L = l[a[i]], R = r[a[i]], t = a[i];
        if(R>i) continue;
        memset(vis, 0, sizeof(vis));
        vis[a[i]] = 1;
        for(int j=i; j>0; --j)//遍历每一个[j, i]
        {
            if(!vis[a[j]])
            {
                vis[a[j]] = 1;
                t ^= a[j];
            }
            L = min(L, l[a[j]]);
            R = max(R, r[a[j]]);
            if(L==j && R==i) dp[i] = max(dp[i], dp[L-1]+t);//如果[j, i]合法, 更新dp[i]
            if(R>i) break;
        }
    }
    cout << dp[n] << endl;

    return 0;
}