【BZOJ】【P3555】【Ctsc2014】【企鹅QQ】【题解】【hash】

传送门:http://www.lydsy.com/JudgeOnline/problem.php?id=3555

史上最水的ctsc题……

枚举不同的位置,hash判断

Code:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<map>
#include<cctype>
using namespace std;
typedef unsigned long long UL;
const int maxn=30010;
UL hash[maxn][202];
UL base=233;
UL hash_l[202];
int n,len;
char s[202];
UL ans=0;
void hash_init(int x){
	UL val=0;
	for(int i=0;i<len;i++)
	val=val*base+s[i],
	hash[x][i]=val;
}
 UL GetHash(int x,int l,int r){
	if(l>r)return 0;
	return hash[x][r]-(l?hash[x][l-1]:0)*hash_l[r-l+1];	
}
pair<UL,UL> H[maxn];
int Hsize;
int main(){
	scanf("%d%d%*d",&n,&len);
	hash_l[0]=1;
	for(int i=1;i<=len;i++)hash_l[i]=hash_l[i-1]*base;
	for(int i=1;i<=n;i++){
		scanf("%s",s);
		hash_init(i);
	}
	for(int i=0;i<len;i++){
		Hsize=0;
		for(int j=1;j<=n;j++)
		H[++Hsize]=make_pair(GetHash(j,0,i-1),GetHash(j,i+1,len-1));
		sort(H+1,H+1+Hsize);
		for(int i=1;i<n;i++){
			int cnt=1;
			while(i!=n&&H[i]==H[i+1])cnt++,i++;
			ans+=(UL)(cnt-1)*cnt/2;
		}
	}cout<<ans<<endl;
	return 0;
}