arc110E Shorten ABC

Description

• 给一个只包含’A’‘B’'C’的字符串$s$，每一次可以选择相邻的两个不相同的字符将它们合成与它们不同的字符。
• 问$s$经过任意次操作可以变成的不同的字符串有多少个
• $|S|\le 1e6$

Solution

• 与agc027E一样
• 假设$ABC$分别是$123$，那么如果不考虑相邻的不同的话，那么操作就是将$x,y$替换为$xxory$，那么显然这个是满足结合律和交换律的。
• 再考虑相邻的相同的情况，显然两个相同的也是可以相消的。
• 讨论一下就会发现只需要把全部相同的情况特判掉，之后就可以转化为不考虑相邻不同的问题来做了。
• 那么从一个$S$变为$T$，判断是否可行我们可以直接贪心，每一次选取最短的段，并且保证最后剩下的段$xor$和为0即可。
• 那么直接DP，设$f[i]$表示1~i的答案，枚举下一个是哪个字符，找到最前的位置转移过去即可。

---

• agc027E是把$ab$看作$12$，在模意义下的加法，需要特判掉不存在相邻相同的情况，其他的证明讨论都是类似的。

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define maxn 1000005
#define mo 1000000007
using namespace std;

int n,i,j,k,a[maxn],f[maxn],s[maxn],nex[maxn][4],bz[4],ans;

int main(){
	freopen("ceshi.in","r",stdin);
	scanf("%d",&n); char ch=getchar();
	while (ch<'A'||ch>'C') ch=getchar();
	for(i=1;i<=n;i++) a[i]=ch-'A'+1,ch=getchar();
	for(k=1;k<=3;k++) {
		for(i=1;i<=n&&a[i]==k;i++);
		if (i>n) printf("1\n"),exit(0);
	}
	for(i=1;i<=n;i++) s[i]=s[i-1]^a[i];
	for(i=0;i<4;i++) bz[i]=n+1;
	for(i=n;i>=0;i--){
		memcpy(nex[i],bz,sizeof(bz));
		bz[s[i]]=i;
	}
	f[0]=1;
	for(i=0;i<=n;i++) if (f[i]){
		if ((s[n]^s[i])==0&&i>0) (ans+=f[i])%=mo;
		for(j=0;j<4;j++) if (j!=s[i]&&nex[i][j]<=n)
			(f[nex[i][j]]+=f[i])%=mo;
	}
	printf("%d\n",ans);
}