【BZOJ 3555】【CTSC 2014】企鹅QQ【hash】

Description

小Q定义，若两个账户名称是相似的，当且仅当这两个字符串等长且恰好只有一位不同。例如“Penguin1”和“Penguin2”是相似的，但“Penguin1”和“2Penguin”不是相似的。而小Q想知道，在给定的 个账户名称中，有多少对是相似的。
为了简化你的工作，小Q给你的 个字符串长度均等于 ，且只包含大小写字母、数字、下划线以及‘@’共64种字符，而且不存在两个相同的账户名称。

Input

第一行包含三个正整数 n，l ，s 。其中 表示账户名称数量， 表示账户名称长度， 用来表示字符集规模大小，它的值只可能为2或64。
若 s 等于2，账户名称中只包含字符‘0’和‘1’共2种字符；
若 s 等于64，账户名称中可能包含大小写字母、数字、下划线以及‘@’共64种字符。
随后 行，每行一个长度为 的字符串，用来描述一个账户名称。数据保证 个字符串是两两不同的。

Output

仅一行一个正整数，表示共有多少对相似的账户名称。

题解

本题要点在枚举删除哪一位，然后对每个字符串hash，在搞个排序，直接统计就可以了。
详可见代码

代码

#include<cstdio>
#include<cstring>
#include<algorithm>

#define N 30010
typedef unsigned long long ll;
using namespace std;
ll h[N][210],g[N][210];
ll tmp[N];
char ch[210];
int ans,n,m,k;

void cal(int x) {
    for(int i = 1;i <= m;i++)
        h[x][i] = (h[x][i-1]*233 + ch[i]);
    for(int i = m;i >= 1;i--)
        g[x][i] = (g[x][i+1]*149 + ch[i]);
}

int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for(int i = 1;i <= n;i++) {
        scanf("%s",ch+1);
        cal(i);
    }
    for(int j = 1;j <= m;j++) {
        for(int i = 1;i <= n;i++)
            tmp[i] = h[i][j-1]*173 + g[i][j+1]*201;
        sort(tmp+1,tmp+n+1);
        int now = 1;
        for(int i = 2;i <= n;i++)
            if(tmp[i] == tmp[i-1]) ans += now,now++;
            else now = 1;
    }
    printf("%d\n",ans);
    return 0;
}