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;
}