BZOJ3555: [Ctsc2014]企鹅QQ（字符串hash）

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pb push_back
#define X first
#define Y second
#define cl(a,b) memset(a,b,sizeof(a))
typedef pair<int,int> P;
const int maxn=30005;
const int inf=1<<27;
const int base=175;//随机选，不会让hash重复就好
int n,m,x;
LL ans;
char s[maxn];
LL a[maxn],sum[maxn][250],power[250];
void cal(int pos){
for(int i=0;i<n;i++){
a[i]=sum[i][m]-sum[i][pos]*power[m-pos]+sum[i][pos-1]*power[m-pos+1];
}
sort(a,a+n);
LL t=0;
for(int i=0;i<n;i++){//C(n,2)==n*(n-1)/2==1+2+3+...+n-1
if(i==0||a[i]!=a[i-1])t=-1;
t++;
ans+=t;
}
}
int main(){
scanf("%d%d%d",&n,&m,&x);
power[0]=1;
for(int i=1;i<250;i++)power[i]=power[i-1]*base;

for(int i=0;i<n;i++){
scanf("%s",&s[1]);
for(int j=1;j<=m;j++){
sum[i][j]=sum[i][j-1]*base+s[j];
}
}
for(int i=1;i<=m;i++){
cal(i);
}
printf("%lld\n",ans);
return 0;
}