一个int不如long long,APhash不如暴力hash,单hash不如双hash的故事……
2017.8.23UPD:上面说的话第二句有问题,因为本人没有搞清APhash的本质导致错误。
CODE:
#include<cstdio>
#include<algorithm>
using namespace std;
#define base1 3131
#define base2 31313
typedef long long ll;
typedef pair<ll,ll> pi;
char s[30005];
ll h1[30005][205],h2[30005][205],g1[30005][205],g2[30005][205];
pi tmp[30005];
int n,len,size,ans;
inline bool operator <(pi a,pi b)
{
if(a.first==b.first) return a.second<b.second;
return a.first<b.first;
}
inline void hash(int id)
{
for(int i=1;i<=len;i++)
h1[id][i]=h1[id][i-1]*base1+s[i],
g1[id][i]=g1[id][i-1]*base2+s[i];
for(int i=len;i;i--)
h2[id][i]=h2[id][i+1]*base2+s[i],
g2[id][i]=g2[id][i+1]*base1+s[i];
}
int main()
{
scanf("%d%d%d",&n,&len,&size);
for(int i=1;i<=n;i++)
scanf("%s",s+1),hash(i);
for(int l=1;l<=len;l++)
{
for(int i=1;i<=n;i++)
tmp[i].first=h1[i][l-1]*base2+h2[i][l+1]*base1,
tmp[i].second=g1[i][l-1]*base2+h2[i][l+1]*base1;
sort(tmp+1,tmp+n+1);
int base=1;
for(int i=2;i<=n;i++)
if(tmp[i]==tmp[i-1]) ans+=base++;
else base=1;
}
printf("%d",ans);
return 0;
}