哈希水题,每次判断去掉这一位的哈希值是否是一样的即可。
选取的质数为137,173,37,73。
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int N = 30010;
const int M = 300;
char ch[M];
unsigned long long H[N][M],G[N][M],tmp[N];
int m,n,k,ans,t=1;
void hash_init(int x){
for(int i=1;i<=m;i++)
H[x][i]=H[x][i-1]*173+ch[i];
for(int i=m;i>=1;i--)
G[x][i]=G[x][i+1]*137+ch[i];
}
int main(){
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++){
scanf("%s",ch+1);
hash_init(i);
}
for(int i=1;i<=m;i++){
for(int j=1;j<=n;j++){
tmp[j]=H[j][i-1]*73+G[j][i+1]*37;
}
sort(tmp+1,tmp+1+n);
for(int j=2;j<=n;++j){
if(tmp[j]==tmp[j-1])ans+=t,t++;
else t=1;
}
}
printf("%d\n",ans);
return 0;
}