【CTSC2014】企鹅QQ

本题在洛谷上的链接：https://www.luogu.org/problemnew/show/P4503

---

 

就我现在的水平而言，看起来真不简单。谁知道他近乎裸题。

本题关键在于判断两个字符串是否只有一个字符不同，最朴素的做法是，枚举每对字符串，再枚举他们的每一位，复杂度是O(n^2*l)的，显然无法满足要求。我们考虑用于字符串比较的哈希，如果求出所有字符串不考虑某一位时的哈希，接下来只需要找到相同的数字就可以了，用排序可以将复杂度降到O(nlogn)，整体复杂度为O(l*nlogn)。怎么求字符串不考虑某一位时的哈希呢，我们可以参考求子串哈希的做法，将某一位“抠”掉。详见代码。

#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;
typedef unsigned long long ull;
const int maxn=3e4+5,maxl=205;
const ull base=233,mod=1e6+18;
int n,l,s,ans;
ull hs[maxn][maxl],fs[maxn],pw[maxl];
char S[maxn][maxl];
int main() {
    scanf("%d%d%d",&n,&l,&s);
    for(int i=1;i<=n;++i) scanf("%s",S[i]);
    pw[0]=1;
    for(int i=1;i<=l;++i) pw[i]=pw[i-1]*base;
    for(int i=1;i<=n;++i) {
        hs[i][0]=S[i][0];
        for(int j=1;j<l;++j) hs[i][j]=hs[i][j-1]*base+S[i][j];
    }
    for(int i=0;i<l;++i) {
        if(!i) for(int j=1;j<=n;++j) fs[j]=hs[j][l-1]-hs[j][0]*pw[l-1];
        else for(int j=1;j<=n;++j) fs[j]=hs[j][l-1]-hs[j][i]*pw[l-i-1]+hs[j][i-1]*pw[l-i-1];
        sort(fs+1,fs+n+1);
        int t=1;
        for(int j=1;j<=n;j++) {
            if(fs[j]!=fs[j-1]) t=1;
            else ans+=t,++t;
        }
    }
    printf("%d",ans);
    return 0;
}