定义形如XXX的串是好的串,在好的串的前提下,若这个串后面的第一个字符和这个串的第一个字符不同,这个串就是有趣的串,否则是无聊的串,求S的所有子串中,有趣的串和无聊的串的个数
枚举XXX中X的长度L,每L个位置设一个关键点,那么一个好的串一定会且仅会经过3个关键点,枚举第一个,计算他和第二个关键点的LCP,LCS,可以得到好的串合法的左端点区间,注意到若一个好的串将他右移后仍然是好的串,那么他一定不是有趣的串,所以我们只要看最右边那个左端点的串是否有趣
code:
#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn = 210000;
const int maxd = 20;
int n;
char str[maxn];
struct Suffix_Array
{
int Fir[maxn],Sec[maxn];
int Sa[maxn],Rank[maxn];
int t[maxn];
void Sort(int a[],int rk[],int sa[],int n,int m)
{
for(int i=0;i<=m;i++) t[i]=0;
for(int i=1;i<=n;i++) t[a[rk[i]]]++;
for(int i=1;i<=m;i++) t[i]+=t[i-1];
for(int i=n;i>=1;i--) sa[t[a[rk[i]]]--]=rk[i];
}
void Get_Sa(int S[],int n,int m)
{
for(int i=1;i<=n;i++) Rank[i]=i;
Sort(S,Rank,Sa,n,m);
Rank[Sa[1]]=1;
for(int i=2;i<=n;i++) Rank[Sa[i]]=Rank[Sa[i-1]]+(S[Sa[i]]!=S[Sa[i-1]]);
int t=1;
while(Rank[Sa[n]]!=n)
{
for(int i=1;i<=n;i++)
{
Fir[i]=Rank[i];
Sec[i]=i+t>n?0:Rank[i+t];
Sa[i]=i;
}
Sort(Sec,Sa,Rank,n,n);
Sort(Fir,Rank,Sa,n,n);
Rank[Sa[1]]=1;
for(int i=2;i<=n;i++) Rank[Sa[i]]=Rank[Sa[i-1]]+
(Fir[Sa[i]]!=Fir[Sa[i-1]]||Sec[Sa[i]]!=Sec[Sa[i-1]]);
t<<=1;
}
}
int Height[maxn];
void Get_Height(int S[])
{
int h=0;
for(int i=1;i<=n;i++)
{
if(h) h--;
if(Rank[i]==1) continue;
while(S[Sa[Rank[i]-1]+h]==S[i+h]) h++;
Height[Rank[i]]=h;
}
}
}SA1,SA2;
struct RMQ
{
int ln[maxn];
int pre[maxn][maxd],suf[maxn][maxd];
void Build(int a[])
{
for(int i=1;(1<<i)<=n;i++) ln[1<<i]=i;
for(int i=1;i<=n;i++) if(!ln[i]) ln[i]=ln[i-1];
for(int i=1;i<=n;i++) pre[i][0]=suf[i][0]=a[i];
for(int j=1;(1<<j)<=n;j++)
{
for(int i=1;i+(1<<j)-1<=n;i++)
suf[i][j]=min(suf[i][j-1],suf[i+(1<<j-1)][j-1]);
for(int i=(1<<j);i<=n;i++)
pre[i][j]=min(pre[i][j-1],pre[i-(1<<j-1)][j-1]);
}
}
int Query(int l,int r)
{
if(l>r) swap(l,r); l++;
int len=ln[r-l+1];
return min(suf[l][len],pre[r][len]);
}
}rmq1,rmq2;
int S1[maxn],S2[maxn];
int main()
{
//freopen("tmp.in","r",stdin);
//freopen("tmp.out","w",stdout);
scanf("%s",str+1); n=strlen(str+1);
for(int i=1;i<=n;i++) S1[i]=str[i]-'a'+1; S1[n+1]=-1;
for(int i=1;i<=n;i++) S2[i]=str[n-i+1]-'a'+1; S2[n+1]=-1;
SA1.Get_Sa(S1,n,26);
SA1.Get_Height(S1);
rmq1.Build(SA1.Height);
SA2.Get_Sa(S2,n,26);
SA2.Get_Height(S2);
rmq2.Build(SA2.Height);
ll sum=0,sum2=0;
for(int l=1;l*3<=n;l++)
{
for(int i=l;i+l<=n;i+=l)
{
int j=i+l;
int cp=rmq2.Query(SA2.Rank[n-i+1],SA2.Rank[n-j+1]),cs=rmq1.Query(SA1.Rank[i],SA1.Rank[j]);
int L=max(i-l+1,i-cp+1),R=min(i,j+cs-3*l);
if(L<=R)
{
sum+=R-L+1;
if(S1[R]!=S1[R+3*l]) sum2++;
}
}
}
printf("%lld %lld\n",sum2,sum-sum2);
return 0;
}