[AGC027E]ABBreviate

Description

给出一个字符串s，每次操作可以选择子串aa变为b，也可以选择子串bb变为a
问能得到的本质不同的字符串数量
|s|<=10^5

Solution

xjb编了7788还是去看题解了_(:з」∠)_
首先，一个字符串能变成一个字符a或b，需要存在一步可行操作，即不为ababa…
然后，一个字符串能变成a还是变成b，可以用(cnta+2*cntb)%3计算，若为0则不能变成一个字符
考虑最后的字符串t，可以把t的每个字符对应回s中的若干个区间，满足以上条件
可以证明每次我们取最短的合法区间，最后拼出来会剩一段，若这一段的和为0则t合法
大概考虑一个和为0的abab可以化在任意一个区间中
然后直接根据上面的判定过程Dp即可

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;

const int N=1e5+5,Mo=1e9+7;

int n,sum[N],f[N],nxt[N][3];
char s[N];

void inc(int &x,int y) {x=x+y>=Mo?x+y-Mo:x+y;}

int main() {
	scanf("%s",s+1);n=strlen(s+1);
	bool ok=0;
	fo(i,1,n-1) if (s[i]==s[i+1]) {ok=1;break;}
	if (!ok) {puts("1");return 0;}
	sum[0]=0;fo(i,1,n) sum[i]=(sum[i-1]+((s[i]=='a')?1:2))%3;
	fo(i,0,2) nxt[n][i]=n+1;
	fd(i,n-1,0) {
		fo(j,0,2) nxt[i][j]=nxt[i+1][j];
		nxt[i][sum[i+1]]=i+1;
	}
	f[0]=1;int ans=0;
	fo(i,0,n) {
		fo(j,1,2) inc(f[nxt[i][(sum[i]+j)%3]],f[i]);
		if (i&&sum[i]==sum[n]) inc(ans,f[i]);
	}
	printf("%d\n",ans);
	return 0;
}