Solution
其实这个东西也不知道是称为思路还是套路……
就是在做这类要在字符串中找相同的两段的题目,有一个这样的做法:枚举串的长度
L
L
L,然后把字符串分段,每
L
L
L个分一段,把每一段的第一个位置记为关键位置,这样每个长度为
L
L
L的子串都会经过且仅经过一个关键位置,然后对每个关键位置
p
p
p和下一个关键位置
p
+
L
p+L
p+L,求前缀的
S
[
1..
p
]
、
S
[
1..
p
+
L
]
S[1..p]、S[1..p+L]
S[1..p]、S[1..p+L]的最长公共后缀和后缀
S
[
p
.
.
n
]
、
S
[
p
+
L
.
.
n
]
S[p..n]、S[p+L..n]
S[p..n]、S[p+L..n]的最长公共前缀,然后就得到了包含
p
p
p和
p
+
L
p+L
p+L的两段相同的串
S
[
l
1
.
.
r
1
]
、
S
[
l
2
.
.
r
2
]
S[l_1..r_1]、S[l_2..r_2]
S[l1..r1]、S[l2..r2],当然要限制这两个串不能包含一个以上关键位置,然后就得到这两串任意一个长度为
L
L
L的对应子串都是相同的,然后就可以做很多事情。
对于这道题目,可以对于每个位置算出以这个位置为起点和终点的形如
A
A
AA
AA的串有多少个,然后最后再算答案,这个可以用上面的东西加上简单的差分实现。
找
L
C
P
、
L
C
S
LCP、LCS
LCP、LCS如果是二分+hash,复杂度为
O
(
n
log
2
n
)
O(n\log^2n)
O(nlog2n),如果用后缀数组,复杂度为
O
(
n
log
n
)
O(n\log n)
O(nlogn)。
Code
#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=60010;
const int inf=2147483647;
const int base=233,mod=1000000007;
int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
return x*f;
}
int n;char str[Maxn];
int Hash[Maxn],Pow[Maxn],s1[Maxn],s2[Maxn];
int g(int l,int r){return(Hash[l]-(LL)Hash[r+1]*Pow[r-l+1]%mod+mod)%mod;}
int lcp(int x,int y)
{
int l=1,r=n-y+1;
while(l<=r)
{
int mid=l+r>>1;
if(g(x,x+mid-1)==g(y,y+mid-1))l=mid+1;
else r=mid-1;
}
return l-1;
}
int lcs(int x,int y)
{
int l=1,r=x;
while(l<=r)
{
int mid=l+r>>1;
if(g(x-mid+1,x)==g(y-mid+1,y))l=mid+1;
else r=mid-1;
}
return l-1;
}
int main()
{
int T=read();
Pow[0]=1;for(int i=1;i<=30000;i++)Pow[i]=(LL)Pow[i-1]*base%mod;
while(T--)
{
LL ans=0;
scanf("%s",str+1);
n=strlen(str+1);
Hash[n+1]=0;for(int i=n;i;i--)Hash[i]=((LL)Hash[i+1]*base%mod+str[i]-'a'+1)%mod;
for(int L=1;L<(n>>1);L++)
{
for(int x=1;x+L<=n;x+=L)
{
int y=x+L;
if(str[x]!=str[y])continue;
int l=min(lcs(x,y),L),r=min(lcp(x,y),L);
if(l+r-1<L)continue;
s1[x-l+1]++,s1[x+r-L+1]--;
s2[x-l+(L<<1)]++,s2[x+r+L]--;
}
}
for(int i=1;i<=n;i++)
{
s1[i]+=s1[i-1],s2[i]+=s2[i-1];
ans+=(LL)s1[i]*s2[i-1];
}
for(int i=0;i<=(n<<1);i++)s1[i]=s2[i]=0;
printf("%lld\n",ans);
}
}