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;
}
BZOJ3555 [Ctsc2014]企鵝QQ 【hash】