社区讨论

求调一维dp

P11233[CSP-S 2024] 染色参与者 3已保存回复 7

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
7 条
当前快照
1 份
快照标识符
@m2ue8f5f
此快照首次捕获于
2024/10/29 19:57
去年
此快照最后确认于
2025/11/04 15:44
4 个月前
查看原帖
CPP
#include<bits/stdc++.h>
using namespace std;
#define int long long
int a[201000][2],n,b[201000],ans,t,v[201000],near[200100],s[200100];
#define ccf(qwe,wer) (b[qwe]==b[wer]?b[qwe]:0)
signed main()
{
	scanf("%lld",&t);
	while(t--)
	{
		memset(a,0,sizeof(a));
		memset(v,0,sizeof(v));
		scanf("%lld",&n);
		for(int i=1;i<=n;++i)scanf("%lld",&b[i]);
		for(int i=1;i<=n;++i)s[i]=s[i-1]+ccf(i-1,i);
		for(int i=1;i<=n;++i)
		{
			near[i]=v[b[i]];
			v[b[i]]=i;
		}
		for(int i=1;i<=n;++i)
		{
			a[i][0]=max(a[i][0],a[i-1][0]+ccf(i-1,i));
			a[i][1]=max(a[i][1],a[i-1][1]+ccf(i-1,i));
			if(near[i])
			{
				a[i][0]=max(a[i][0],a[near[i]][0]+s[i-1]-s[near[i]]+b[i]);
				a[i][1]=max(a[i][1],a[near[i]][1]+s[i-1]-s[near[i]]+b[i]);
			}
		}
		printf("%lld\n",max(a[n][0],a[n][1]));
	}
	return 0;
}

回复

7 条回复,欢迎继续交流。

正在加载回复...