【bzoj3555】[Ctsc2014]企鹅QQ hash

题解

其实代码是看hzwer博客的qwq
我们枚举去掉某一位的字符串的情况进行比较。pre[i][j]表示第i个字符串的前j个的hash值，
suf[i][j]表示第i个字符串后j个的hash值。我们计算第i个字符串去掉第j位的hash值就是pre[i][j-1]+suf[i][j+1]。再排序统计一下即可得到答案。

代码

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cctype>
#define ull unsigned long long
#define N 30050
using namespace std;
char s[210];
ull pre[N][210],suf[N][210],cur[N];
int n,m,v,ans;
void init(int x)
{
    for (int i=1;i<=m;i++) pre[x][i]=pre[x][i-1]*131+s[i];
    for (int i=m;i>=1;i--) suf[x][i]=suf[x][i+1]*137+s[i]; 
}
int main()
{
    scanf("%d%d%d\n",&n,&m,&v);
    for (int i=1;i<=n;i++) gets(s+1),init(i);
    for (int i=1;i<=m;i++)
    {
        for (int j=1;j<=n;j++) cur[j]=(pre[j][i-1]*213+suf[j][i+1]*233);
        sort(cur+1,cur+1+n);
        int temp=1;
        for (int j=2;j<=n;j++)
            if (cur[j]==cur[j-1]) ans=ans+temp++;
            else temp=1;
    }
    printf("%d\n",ans);
    return 0;
}