bzoj3555 [Ctsc2014]企鹅QQ

传送门

一个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;
}