【bzoj 3555】[Ctsc2014]企鹅QQ

传送门

分析

求去掉一个字符多少字符串相等。
和friends有点像，只是多个字符串在比较。我们可以枚举删除的点，把剩下的hash值用一个数组来存，把数组排个序，相同的hash值肯定排在一起，加上前面与他相等的即可。

代码

#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
#define full(a,b) memset(a,b,sizeof a)
#define ll long long
#define ull unsigned ll
int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9') f=ch=='-'?-1:1,ch=getchar();
	while(ch>='0'&&ch<='9') x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
	return x*f;
}
#define debug 1
#define N 30005
#define M 205
#define b 131
int n,m;char arr[M];
ull Hash[N][M],p[M],tmp[N];
ll ans;
ull hh(int S,int E,int i)//求第i个字符串中S~E的hash值 
{
	if(S>E) return 0;
	if(!S) return Hash[i][E];
	return Hash[i][E]-Hash[i][S-1]*p[E-S+1];
}
int main()
{
	if(debug==-1)
	{
		freopen("企鹅QQ.in","r",stdin);
		freopen("企鹅QQ.out","w",stdout);
	}
	p[0]=1;
	for(int i=1; i<M; i++)
		p[i]=p[i-1]*b;
	n=read(),m=read(),read();
	for(int i=1; i<=n; i++)
	{
		scanf("\n%s",arr);
		Hash[i][0]=arr[0];
		for(int j=1; j<m; j++)
			Hash[i][j]=Hash[i][j-1]*b+arr[j];
	}
	for(int i=0; i<m; i++)
	{
		for(int j=1; j<=n; j++)
			tmp[j]=hh(0,i-1,j)*p[m-i]+hh(i+1,m-1,j);//求剩余部分的hash值 
		sort(tmp+1,tmp+1+n);
		ll f=1;
		for(int j=2; j<=n; j++)
		{
			if(tmp[j]==tmp[j-1]) ans+=f,f++;//与前面相等，加上前面的 
			else f=1;
		}
	}
	printf("%lld",ans);
	return 0;
}