题目描述
PenguinQQ是中国最大、最具影响力的SNS(Social Networking Services)网站,以实名制为基础,为用户提供日志、群、即时通讯、相册、集市等丰富强大的互联网功能体验,满足用户对社交、资讯、娱乐、交易等多方面的需求。
小Q是PenguinQQ网站的管理员,他最近在进行一项有趣的研究——哪些账户是同一个人注册的。经过长时间的分析,小Q发现同一个人注册的账户名称总是很相似的,例如Penguin1,Penguin2,Penguin3……于是小Q决定先对这种相似的情形进行统计。
小Q定义,若两个账户名称是相似的,当且仅当这两个字符串等长且恰好只有一位不同。例如“Penguin1”和“Penguin2”是相似的,但“Penguin1”和“2Penguin”不是相似的。而小Q想知道,在给定的 个账户名称中,有多少对是相似的。
为了简化你的工作,小Q给你的n个字符串长度均等于m,且只包含大小写字母、数字、下划线以及‘@’共64种字符,而且不存在两个相同的账户名称。
输入
第一行包含三个正整数 n,m,s 。其中n表示账户名称数量,m表示账户名称长度,s用来表示字符集规模大小,它的值只可能为2或64。
若s等于2,账户名称中只包含字符‘0’和‘1’共2种字符;
若s等于64,账户名称中可能包含大小写字母、数字、下划线以及‘@’共64种字符。
随后n行,每行一个长度为m的字符串,用来描述一个账户名称。数据保证n个字符串是两两不同的。
输出
仅一行一个正整数,表示共有多少对相似的账户名称。
样例输入
4 3 64
Fax
fax
max
mac
样例输出
4
题解
字符串hash
先预处理每个字符串的前缀的hash值,然后枚举多余字符的位置,计算出每个字符串剩下串的hash值,判断有多少个相同即可。
注意不要把n和m弄混。
注意不要把n和m弄混。
注意不要把n和m弄混。
#include <cstdio> #include <cstring> #include <algorithm> #define N 30010 #define M 210 using namespace std; unsigned long long hash[N][M] , base[M] , a[N]; int v[N]; char str[M]; int main() { int n , m , i , j , top; long long ans = 0; scanf("%d%d%*d" , &n , &m); for(i = 1 ; i <= n ; i ++ ) { scanf("%s" , str + 1); for(j = 1 ; j <= m ; j ++ ) hash[i][j] = hash[i][j - 1] * 131 + str[j]; } base[0] = 1; for(i = 1 ; i <= m ; i ++ ) base[i] = base[i - 1] * 131; for(i = 1 ; i <= m ; i ++ ) { for(j = 1 ; j <= n ; j ++ ) a[j] = hash[j][m] - hash[j][i] * base[m - i] + hash[j][i - 1] * base[m - i]; sort(a + 1 , a + n + 1); memset(v , 0 , sizeof(v)) , top = 1 , v[1] = 1; for(j = 2 ; j <= n ; j ++ ) { if(a[j] != a[j - 1]) top ++ ; v[top] ++ ; } for(j = 1 ; j <= top ; j ++ ) ans += (long long)v[j] * (v[j] - 1) / 2; } printf("%lld\n" , ans); return 0; }