字符串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=30000+500,maxl=250;
int N,L,S,lA,lB,base=13331;
ull power[maxl],mo=1e9+7,sum[maxn][maxl],h[maxn],H[maxn];
ll A,B,ans=0,temp;
char str[maxn][maxl];
ull hash(int i,int x,int y){
    return (sum[i][y]-sum[i][x-1]*power[y-x+1]);
}
int main(){
    scanf("%d%d%d",&N,&L,&S);    
    power[0]=1;
    for(int i=1;i<=205;i++){
        power[i]=power[i-1]*base;
    }
    
    for(int i=1;i<=N;i++)scanf("%s",str[i]+1);
    for(int i=1;i<=N;i++){
        for(int j=1;j<=L;j++){
            sum[i][j]=sum[i][j-1]*base+str[i][j];
        }
    }
    for(int i=1;i<=L;i++){
        for(int j=1;j<=N;j++){
            if(i==1){
                H[j]=hash(j,2,L);
            }
            else if(i==L){
                H[j]=hash(j,1,L-1);
            }
            else{
                A=hash(j,1,i-1);
                B=hash(j,i+1,L);
                H[j]=A*power[L-i]+B;
            }
        }
        sort(H+1,H+N+1);
        temp=1;
        for(int j=2;j<=N;j++){
            if(H[j]==H[j-1]){
                ans+=temp;
                temp++;
            }
            else temp=1;
        }
    }
    printf("%lld",ans);
    return 0;
}

Tips:

字符串hash拼接公式：

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

---

By:AlenaNuna