Problem - C - Codeforces
Tenzing在一条线上排列了n个球。从左边起第i个球的颜色是ai。
Tenzing可以进行以下操作任意次:
选择i和j,使得1≤i<j≤|a|且ai=aj,
从数组中删除ai,ai+1,…,aj(并将所有元素的索引向右移动aj−i+1)。
Tenzing想知道他能够删除的最多的球数量。 输入格式
每个测试包含多个测试用例。第一行输入一个整数t(1≤t≤10^3)——测试用例的数量。接下来描述每个测试用例。
第一行包含一个整数n(1≤n≤2⋅10^5)——球的数量。
第二行包含n个整数a1,a2,…,an(1≤ai≤n)——球的颜色。
保证所有测试用例n的总和不超过2*10^5。 输出格式
对于每个测试用例,输出Tenzing可以删除的最大球数。 样例
Example
Input
Copy
2
5
1 2 2 3 3
4
1 2 1 2
Output
Copy
4 3
在第一个示例中,Tenzing将在第一次操作中选择i=2和j=3,使得a=[1,3,3]。然后Tenzing再次选择i=2和j=3,使得a=[1]。因此Tenzing总共可以删除4个球。
在第二个示例中,Tenzing将在唯一的操作中选择i=1和j=3,使得a=[2]。因此Tenzing总共可以删除3个球。
题解:
简单dp,转移方程
对于当前位置i,之前出现过a[i]的位置j
dp[i] = dp[i - 1] + i - j + 1;
但是转化为代码,好像题目并没有体现出这个式子
但是其实非常简单,只是转化了一些
dp[i] = i + (dp[i - 1] - (j - 1))
这行代码就代表后面这一部分(dp[i - 1] - (j - 1))
#include <cstdio>
#include <cstring>
#include <algorithm>
#include<iostream>
#include<vector>
#include<set>
#include<map>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
#define int long long
typedef pair<int,int> PII;
typedef unsigned long long ULL;
const int N = 4e5 + 10;
int mod = 998244353;
void solve()
{
int n;
cin >> n;
vector<int> dp(n + 10),mx(n + 10,-1e9),a(n + 10);
for(int i = 1;i <= n;i++)
{
cin >> a[i];
}
for(int i = 1;i <= n;i++)
{
dp[i] = dp[i - 1];
dp[i] = max(dp[i],mx[a[i]] + i);
mx[a[i]] = max(mx[a[i]],dp[i - 1] - (i - 1));
}
cout << dp[n] <<"\n";
}
signed main()
{
ios::sync_with_stdio(0 );
cin.tie(0);cout.tie(0);
int t = 1;
cin >> t;
while(t--)
{
solve();
}
}