字符串Hash || BZOJ 3555: [Ctsc2014]企鹅QQ || P4503 [CTSC2014]企鹅QQ

题面：[CTSC2014]企鹅QQ

题解：无

代码：

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #include<cstdlib>
 #include<cmath>
 #include<map>
 #include<algorithm>
 #define ll long long
 #define max(a,b) ((a)>(b)?(a):(b))
 #define min(a,b) ((a)<(b)?(a):(b))
 #define fo(i,a,b) for(int i=a;i<=b;i++)
 #define fd(i,a,b) for(int i=a;i>=b;i--)
 typedef unsigned long long ull;
 using namespace std;
 const int maxn=+,maxl=;
 int N,L,S,lA,lB,base=;
 ull power[maxl],mo=1e9+,sum[maxn][maxl],h[maxn],H[maxn];
 ll A,B,ans=,temp;
 char str[maxn][maxl];
 ull hash(int i,int x,int y){
     return (sum[i][y]-sum[i][x-]*power[y-x+]);
 }
 int main(){
     scanf("%d%d%d",&N,&L,&S);
     power[]=;
     for(int i=;i<=;i++){
         power[i]=power[i-]*base;
     }

     for(int i=;i<=N;i++)scanf("%s",str[i]+);
     for(int i=;i<=N;i++){
         for(int j=;j<=L;j++){
             sum[i][j]=sum[i][j-]*base+str[i][j];
         }
     }
     for(int i=;i<=L;i++){
         for(int j=;j<=N;j++){
             if(i==){
                 H[j]=hash(j,,L);
             }
             else if(i==L){
                 H[j]=hash(j,,L-);
             }
             else{
                 A=hash(j,,i-);
                 B=hash(j,i+,L);
                 H[j]=A*power[L-i]+B;
             }
         }
         sort(H+,H+N+);
         temp=;
         for(int j=;j<=N;j++){
             if(H[j]==H[j-]){
                 ans+=temp;
                 temp++;
             }
             else temp=;
         }
     }
     printf("%lld",ans);
     return ;
 }

Tips:

字符串hash拼接公式：

拼接串a和串b：hash[a]*(power[lenb])+hash[b]

---

By:AlenaNuna