【题解】AtCoder ARC128D - Neq Neq

题意：有 $N$ 个球排成一行，编号为 $1$ 到 $N$ 。 第 $i$个球上写着一个数$A_i$。

你可以执行下列操作任意多次：

• 选三个相邻的位置 $x$, $y$, $z$ $(1\le x<y<z\le N)$ ，满足 $A_x\ne A_y$ 并且 $A_y\ne A_z$。
• 删掉 $A_y$, $A_x$ 就和 $A_z$ 连在一起了。

问最后剩余的球有多少种不同的方案数，模 $998244353$ 。

题解:

$d{p}_i$ 表示当前操作到第 $i$ 个位置的方案数，如果有这样一种情况: $1234456$ ，其中 $44$ 永远不会被删掉。不难想到，如果有两个相邻的数相等，则前面的 $dp$ 值不可能转移到后面去。根据样例 $1212$ 的情况，猜想一个数之前有三个以上不同的数才能删到当前这个数，于是用前缀和优化。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll MOD=998244353;
int n;
int cnt[200010];
ll a[200010],dp[200010],sum[200010];
int main(){
	cin>>n;
	int num=0;
	int j=0,k=1;
	dp[0]=1;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		dp[i]=dp[i-1];
		if(a[i]==a[i-1])j=i-1;
		if(i>2&&a[i]!=a[i-1]&&a[i-1]!=a[i-2])dp[i]=(dp[i]+dp[i-2])%MOD; 
		if(!cnt[a[i]])num++;
		cnt[a[i]]++;
		while(k+2<i&&num>=3){
			cnt[a[k]]--;
			if(cnt[a[k]]==0)num--;
			k++;
		}
		if(j<k)dp[i]=(dp[i]+sum[k-1]-sum[j])%MOD;
		sum[i]=(sum[i-1]+dp[i])%MOD;
	}
	cout<<dp[n]<<endl;
	return 0;
}