字符串hash——洛谷P1117 优秀的拆分

#include<bits/stdc++.h>
#define pow fuck
#define Ll long long
using namespace std;
const int N=2e3+5;
Ll A
,B
,hush
,pow
;
Ll S=999983,mo=1e8+9,ans;
int n,m;
char c
;
Ll get(int x,int y){return (hush[y]-hush[x-1]*pow[y-x+1]%mo+mo)%mo;}
int main()
{
scanf("%d",&m);
pow[0]=1;
for(int i=1;i<=N;i++)pow[i]=pow[i-1]*S%mo;
while(m--){
scanf("%s",c+1);
n=strlen(c+1);
for(int i=1;i<=n;i++)hush[i]=(hush[i-1]*S+c[i]-96)%mo;
memset(A,0,sizeof A); memset(B,0,sizeof B);
for(int i=1;i<=n;i++)
for(int j=i;j+j-i-1<=n;j++)
if(get(i,j)==get(j+1,j+j-i+1))
A[i]++,B[j+j-i+1]++;
ans=0;
for(int i=1;i<n;i++)ans+=B[i]*A[i+1];
printf("%lld\n",ans);
}
}