【bzoj3555】[Ctsc2014]企鹅QQ 简单哈希

传送门

题目分析

题意即求有多少对字符串只相差一个字符，枚举删除每个字符后的哈希， 看有多少相等即可。

比如有如下字符串：$Sd123$，其中S部分的哈希值为H，删除的是d，则原字符串的哈希值为$$(((H * T + d) * T + 1) * T + 2) * T + 3 = H * T^4 + d * T^3 + 1 * T^2 + 2 * T + 3$$

删除过后就为$$((H * T + 1) * T + 2) * T +3 = H * T^3 + 1 * T^2 + 2 * T + 3$$

也就是将1以前的哈希值全部剪去然后加上d之前的哈希值乘以$T^{len - delpos}$。

code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;

const int N = 3e4 + , L = , Mod = ;
typedef unsigned long long ull;
const ull H = ;
ull hash[N][L], poww[L], go[N << ], t[N];
int ecnt, adj[Mod + ], nxt[N << ];
int n, l, m, cnt;
char s[N][L];

inline ull del(int k, int d){
    return hash[k][l] - hash[k][d] * poww[l - d] + hash[k][d - ] * poww[l - d];
}

int main(){
    scanf("%d%d%d", &n, &l, &m);
    for(int i = ; i <= n; i++)
        scanf("%s", s[i] + );
    for(int i = ; i <= n; i++)
        for(int j = ; j <= l; j++)
            hash[i][j] = hash[i][j - ] * H + s[i][j];
    poww[] = ;
    for(int i = ; i <= L; i++) poww[i] = poww[i - ] * H;
    for(int j = ; j <= l; j++){
        for(int i = ; i <= n; i++)
            t[i] = del(i, j);
        sort(t + ,t + n + );
        int sum = ;
        for(int i = ; i <= n; i++){
            if(t[i] == t[i - ]){
                cnt += sum;
                sum++;
            }
            else sum = ;
        }
    }
    printf("%d", cnt);
    return ;
}