bzoj3555 [Ctsc2014]企鹅QQ hash

Description

---

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

N<=30000,L<=200,S<=64

Solution

---

考虑枚举不同的那一位，用hash可以很方便写

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

typedef unsigned long long LL;
const int N=30005;
const int L=205;

char str[N][L];

LL h[N][L],p[N],a[N];

int main(void) {
    int n,m,k; scanf("%d%d%d",&n,&m,&k);
    p[0]=1; rep(i,1,m) p[i]=p[i-1]*149;
    rep(i,1,n) {
        scanf("%s",str[i]+1);
        rep(j,1,m) h[i][j]=h[i][j-1]*149+str[i][j];
    }
    LL ans=0;
    rep(i,1,m) {
        rep(j,1,n) a[j]=h[j][m]-h[j][i]*p[m-i]+h[j][i-1]*p[m-i+1];
        std:: sort(a+1,a+n+1);
        LL s=1;
        rep(j,2,n) if (a[j]!=a[j-1]) {
            ans+=s*(s-1)/2; s=1;
        } else s++;
        ans+=s*(s-1)/2;
    }
    printf("%lld\n", ans);
    return 0;
}