题目
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;
}