CF1842C题解

CF1842C

Lougu题面

CF Problem

CF Contest

题目描述

共有 $t$ 组测试样例。

对于每组测试样例，有一个大小为 $n$ 的数组 $a$。你可以进行下列一组操作任意多次：

• 选择 $i$ 和 $j$，使$1\le i<j\le |a|$ 并且$a_i=a_j$，
• 从数组中删除 $a_i,a_{i+1},\dots ,a_j$（并将 $a_j$ 右边的所有元素的索引减少 $j-i+1$）。

求出能删除数的最大数量。

注：$|a|$ 表示 $a$ 数组的大小。

数据范围

$1\le t\le 10^3$

$1\le n\le 2\cdot 10^5$

$1\le a_i\le n$

保证所有测试样例的 $n$ 总和不超过 $2\cdot 10^5$ 。

思路

对于四个数组下标 $a,b,c,d$ 满足 $a<b<c<d$ :

• 首先，不可能同时删除范围 $(a,c)$ 和 $(b,d)$ 。
• 其次，如果我们删除范围 $(a,d)$ 和 $(b,c)$ ，我们必须先删除范围 $(b,c)$ ，然后再删除范围 $(a,d)$ 。然而，先删除范围 $(b,c)$ 再删除 $(a,d)$ 与只删除 $(a,d)$ 的效果是一样的。

因此，我们可以假设，在最优解中，删除的范围都是不相交的。题目就转化为：希望找到某几个不相交的范围 $[l_1,r_1],[l_2,r_2],\dots ,[l_m,r_m]$ ，使得 $a_{l_i}=a_{r_i}$ 且 ${\sum }_{i=1}^m(r_i-l_i+1)$ 为最大值。

我们可以写一个 DP 。$d{p}_i$ 表示数组 $a$ 前 $i$ 位删除的最大球数。

可以 $O(n^2)$ 枚举由第 $j$ 位转移，得到转移方程：

$$d{p}_i=\max (d{p}_{i-1},d{p}_{j-1}+i-j+1)$$

考虑优化 DP 。观察到 DP 转移方程中的未知量仅有 $d{p}_{j-1}-j+1$，由于 $1\le a_i\le n$ ，可以使用一个桶数组 $s$ 记录对于数字 $x$ ，满足 $a_j=x$ 最大的 $d{p}_{j-1}-j+1$ 。

注意：$s[i]$ 为负数，初始化需设为极小值。

这样就将 DP 转移方程优化到了 $O(n)$ ：

$$\begin{gathered} d{p}_i=\max (d{p}_{i-1},s_{a_i}+i) \\ {s}_{a_i}=\max (s_{a_i},d{p}_{i-1}-i+1) \end{gathered}$$

代码

#include<bits/stdc++.h>
using namespace std;

int t,n;
int a[200010],dp[200010],s[200010];

int main()
{
    scanf("%d",&t);
	while(t--)
	{
		int ans=0;
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
			s[i]=-1e9;//初始化为极小值
		for(int i=1;i<=n;i++)
			scanf("%d",&a[i]);
		for(int i=1;i<=n;i++)
		{
			dp[i]=max(dp[i-1],s[a[i]]+i);//DP转移方程
			s[a[i]]=max(s[a[i]],dp[i-1]-i+1);		    
		}
		printf("%d\n",dp[n]);
	}
	return 0 ;
}