bzoj3555: [Ctsc2014]企鹅QQ hash

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=30005;
const int M=205;
const int mo1=23333333;
const int mo2=1e9+7;
int n,m,k;
typedef long long ll;
ll a[N],pre[N][M],suf[N][M],ans;
char s[N][M];
inline void cal(int x)
{
fo(i,1,m)
pre[x][i]=(pre[x][i-1]*211+s[x][i]);
fd(i,m,1)
suf[x][i]=(suf[x][i+1]*179+s[x][i]);
}
int main()
{
scanf("%d%d%d",&n,&m,&k);
fo(i,1,n)
{
scanf("%s",s[i]+1);
cal(i);
}
fo(j,1,m)
{
fo(i,1,n)
{
a[i]=(pre[i][j-1]*311+suf[i][j+1]*317);
}
sort(a+1,a+1+n);
int cnt=1;
fo(i,1,n)
{
if (a[i]==a[i-1])ans+=cnt,cnt++;
else cnt=1;
}
}
printf("%lld\n",ans);
}