C. Tenzing and Balls(dp)

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(); 
	}
}