【BZOJ3555】企鹅QQ(字符串哈希)

题面

BZOJ

题解

把前缀哈希一下,后缀哈希一下
枚举哪个位置不选,然后检查一下相同就行了。。
为什么我的 Hash 老是 WA
为什么 Zsy 一遍 AC 。、。
无奈啊。。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ull unsigned long long
#define RG register
int n,L,ans,sig;
ull pre[30003][222],suf[30003][222],Suf[30003][222],S[30003];
char T[222];
int main()
{
    scanf("%d%d%d",&n,&L,&sig);
    for(int i=1;i<=n;++i)
    {
        scanf("%s",T+1);
        for(int j=1;j<=L;++j)pre[i][j]=pre[i][j-1]*19260817+T[j];
        for(int j=L;j>=1;--j)suf[i][j]=suf[i][j+1]*998244353+T[j];
        for(int j=L;j>=1;--j)Suf[i][j]=Suf[i][j+1]*1000000007+T[j];
    }
    for(int i=1;i<=L;++i)
    {
        for(int j=1;j<=n;++j)S[j]=pre[j][i-1]*1007+suf[j][i+1]*197+Suf[j][i+1];
        sort(&S[1],&S[n+1]);
        for(int j=2,s=1;j<=n;++j)
            if(S[j]==S[j-1])ans+=s++;
            else s=1;
    }
    printf("%d\n",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_30974369/article/details/79930923