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对相似的字符串分别为：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;

}