BZOJ3555 [Ctsc2014]企鹅QQ 【hash】

题目

PenguinQQ是中国最大、最具影响力的SNS（Social Networking Services）网站，以实名制为基础，为用户提供日志、群、即时通讯、相册、集市等丰富强大的互联网功能体验，满足用户对社交、资讯、娱乐、交易等多方面的需求。
小Q是PenguinQQ网站的管理员，他最近在进行一项有趣的研究——哪些账户是同一个人注册的。经过长时间的分析，小Q发现同一个人注册的账户名称总是很相似的，例如Penguin1，Penguin2，Penguin3……于是小Q决定先对这种相似的情形进行统计。
小Q定义，若两个账户名称是相似的，当且仅当这两个字符串等长且恰好只有一位不同。例如“Penguin1”和“Penguin2”是相似的，但“Penguin1”和“2Penguin”不是相似的。而小Q想知道，在给定的 个账户名称中，有多少对是相似的。
为了简化你的工作，小Q给你的 个字符串长度均等于 ，且只包含大小写字母、数字、下划线以及‘@’共64种字符，而且不存在两个相同的账户名称。

输入格式

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

输出格式

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

输入样例

4 3 64

Fax

fax

max

mac

输出样例

4

提示

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

题解

字符串长度不大，我们考虑每次求删去同一个位置后相同的字符串数量
这样做加上hash就是\(O(n*len)\)的

当然每次不能暴力求出hash值
应先求出原来字符串的hash值，每次\(O(n)\)算出新的就行了

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define LL long long int
#define Redge(u) for (int k = h[u]; k; k = ed[k].nxt)
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define ULL unsigned long long int
using namespace std;
const int maxn = 30005,maxm = 205,INF = 1000000000;
inline int read(){
	int out = 0,flag = 1; char c = getchar();
	while (c < 48 || c > 57){if (c == '-') flag = -1; c = getchar();}
	while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
	return out * flag;
}
int n,len;
ULL H[maxn][maxm],A[maxn];
char s[maxn][maxm];
int id(char c){
	if (c >= 'a' && c <= 'z') return c - 'a' + 1;
	if (c >= 'A' && c <= 'Z') return c - 'A' + 27;
	if (c >= '0' && c <= '9') return c - '0' + 53;
	if (c == '_') return 63;
	return 64;
}
LL C(LL x){return x * (x - 1) / 2;}
int main(){
	n = read(); len = read(); read();
	for (int i = 1; i <= n; i++){
		scanf("%s",s[i] + 1);
		for (int j = 1; j <= len; j++)
			H[i][j] = H[i][j - 1] * 67 + id(s[i][j]);
	}
	LL ans = 0;
	for (int i = 1; i <= len; i++){
		ULL P = 1,PP = 1;
		for (int j = i + 1; j <= len; j++) PP *= 67;
		for (int j = 1; j < i; j++) P *= 67;
		for (int j = 1; j <= n; j++)
			A[j] = (H[j][len] - H[j][i] * PP) * P + H[j][i - 1];
		sort(A + 1,A + 1 + n);
		int cnt = 0;
		for (int j = 1; j <= n; j++){
			if (j == 1 || A[j] != A[j - 1]) cnt = 0;
			cnt++;
			if (j == n || A[j] != A[j + 1])
				ans += C(cnt);
		}
	}
	printf("%lld\n",ans);
	return 0;
}