BZOJ 3555: [Ctsc2014]企鹅QQ

Description

PenguinQQ是中国最大、最具影响力的SNS（Social Networking Services）网站，以实名制为基础，为用户提供日志、群、即时通讯、相册、集市等丰富强大的互联网功能体验，满足用户对社交、资讯、娱乐、交易等多方面的需求。

小Q是PenguinQQ网站的管理员，他最近在进行一项有趣的研究——哪些账户是同一个人注册的。经过长时间的分析，小Q发现同一个人注册的账户名称总是很相似的，例如Penguin1，Penguin2，Penguin3……于是小Q决定先对这种相似的情形进行统计。

小Q定义，若两个账户名称是相似的，当且仅当这两个字符串等长且恰好只有一位不同。例如“Penguin1”和“Penguin2”是相似的，但“Penguin1”和“2Penguin”不是相似的。而小Q想知道，在给定的 个账户名称中，有多少对是相似的。

为了简化你的工作，小Q给你的 个字符串长度均等于 ，且只包含大小写字母、数字、下划线以及‘@’共64种字符，而且不存在两个相同的账户名称。

Input

第一行包含三个正整数 ， ， 。其中 表示账户名称数量， 表示账户名称长度， 用来表示字符集规模大小，它的值只可能为2或64。

若 等于2，账户名称中只包含字符‘0’和‘1’共2种字符；

若 等于64，账户名称中可能包含大小写字母、数字、下划线以及‘@’共64种字符。

随后 行，每行一个长度为 的字符串，用来描述一个账户名称。数据保证 个字符串是两两不同的。

Output

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

Sample Input

4 3 64

Fax

fax

max

mac

Sample Output

4

HINT

4对相似的字符串分别为：Fax与fax，Fax与max，fax与max，max与mac。N<=30000,L<=200,S<=64

分析

枚举删去每一位，然后哈希判断就好

代码

#include <bits/stdc++.h>

#define ll unsigned long long
#define N 30005
#define M 205
#define inf 2000000000

int read()
{
int x = 0, f = 1;
char ch = getchar();
while (ch < '0' || ch > '9') {if (ch == '-') f = -1; ch = getchar();}
while (ch >= '0' && ch <= '9') {x = x * 10 + ch - '0'; ch = getchar();}
return x * f;
}

char ch[M];
ll f
[M],g
[M];
ll tmp
;

int n,m,K;

void getHash(int x)
{
for (int i = 1; i <= m; i++)
f[x][i] = f[x][i - 1] * 149 + ch[i];
for (int i = m; i >= 1; i--)
g[x][i] = g[x][i + 1] * 137 + ch[i];
}

int main()
{
n = read(), m = read(), K = read();
for (int i = 1; i <= n; i++)
{
scanf("%s", ch + 1);
getHash(i);
}
int ans = 0;
for (int j = 1; j <= m; j++)
{
for (int i = 1; i <= n; i++)
tmp[i] = f[i][j - 1] * 233 + g[i][j + 1] * 213;
std::sort(tmp + 1, tmp + n + 1);
int cnt = 1;
for (int i = 2; i <= n; i++)
{
if (tmp[i] == tmp[i - 1])
ans += cnt, cnt++;
else cnt = 1;
}
}
printf("%d\n",ans);
}