【bzoj 3555】企鹅QQ（字符串哈希）

传送门biu~
枚举不看字符串的每一位，用字符串哈希匹配。但是单模会WA，改成双模才能过。

#include<bits/stdc++.h>
#define N 30005
#define base1 137
#define base2 19260817
#define pa pair<int,int>
using namespace std;
int n,ans,l,op,Hash1[N],Hash2[N],Pow1,Pow2;
map<pa,int>p;char s[N][205];
int main(){
    scanf("%d%d%d",&n,&l,&op);
    Pow1=1;for(int i=1;i<=l-2;++i) Pow1*=base1;
    Pow2=1;for(int i=1;i<=l-2;++i) Pow2*=base2;
    for(int i=1;i<=n;++i){
        scanf("%s",s[i]+1);
        for(int j=2;j<=l;++j)   Hash1[i]=Hash1[i]*base1+s[i][j];
        for(int j=2;j<=l;++j)   Hash2[i]=Hash2[i]*base2+s[i][j];
        ans+=p[pa(Hash1[i],Hash2[i])]++;
    }
    for(int i=2;i<=l;++i){
        p.clear();
        for(int j=1;j<=n;++j){
            Hash1[j]=(Hash1[j]-s[j][i]*Pow1)*base1+s[j][i-1];
            Hash2[j]=(Hash2[j]-s[j][i]*Pow2)*base2+s[j][i-1];
            ans+=p[pa(Hash1[j],Hash2[j])]++;
        }
    }
    printf("%d",ans);
    return 0;
}