C. Tenzing and Balls

Problem - 1842C - Codeforces

题目大意：

给定一个序列，你可以拿走任何两个相等元素中间的所有元素，问最多可以拿走多少元素

思路：

dp:

第一次优化：

对于每个元素作为最后一个元素，考虑在它前面和它相等的元素选一个消去中间元素加上前面一个元素的dp值，

\(dp[i]=max(dp[j-1]+i-j+1,dp[i-1]),a[i]=a[j]\)

第二次优化：

对于每个和\(a[i]\)相等位置\(j\)，都要计算\(dp[j-1]+i-j+1\)，然后循环，我们可以优化只保存\(max(dp[j-1]-j+1），a[i]=a[j]\)的最大值

code:

int n, a[N];
void solved()
{
    cin >> n;
    for (int i = 1; i <= n; ++i)
        cin >> a[i];
    vector<int> pos(n + 1, 0);
    int dp[N] = {0},vis[N] = {0};
    for (int i = 1; i <= n; ++i)
    {
        if (vis[a[i]] == 0)
        {
            dp[i] = dp[i - 1];
            pos[a[i]] = dp[i - 1] - i + 1;
            vis[a[i]] = 1;
        }
        else
        {
            dp[i] = max(dp[i - 1], pos[a[i]] + i);
            pos[a[i]] = max(pos[a[i]], dp[i - 1] - i + 1);
        }
    }
    cout << *max_element(dp + 1, dp + n + 1) << endl;
}