bzoj4084【SDOI2015】bigyration

感觉字符串题的做法都好神奇…以后一定要优先考虑哈希

假设S是长度较长的集合。

将T中的串哈希，用map记录每个哈希值出现了多少次。

然后枚举S串，在map中找可以拼接的T串的个数。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<map>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define ull unsigned long long
#define maxn 4000005
#define base 233
using namespace std;
int n,m,len1,len2;
ull p[maxn],f[maxn];
ll ans;
string s[maxn],t[maxn];
map<ll,int> mp;
inline ull hash(int l,int r)
{
	if (l>r) return 0;
	return f[r]-f[l-1]*p[r-l+1];
}
int main()
{
	scanf("%d%d%d%d",&n,&m,&len1,&len2);
	p[0]=1;
	F(i,1,len1+len2) p[i]=p[i-1]*base;
	int mid=(len1+len2)>>1;
	if (len1<len2)
	{
		F(i,1,n) cin>>t[i];
		F(i,1,m) cin>>s[i];
		swap(n,m);swap(len1,len2);
	}
	else
	{
		F(i,1,n) cin>>s[i];
		F(i,1,m) cin>>t[i];
	}
	F(i,1,m)
	{
		ull h=0;
		F(j,0,len2-1) h=h*base+t[i][j]-'a'+1;
		mp[h]++;
	}
	F(i,1,n)
	{
		F(j,0,mid*2-1) f[j+1]=f[j]*base+s[i][j%mid]-'a'+1;
		ull h=0;
		F(j,mid,len1-1) h=h*base+s[i][j]-'a'+1;
		F(j,1,mid) if (hash(j,j+len1-mid-1)==h) ans+=mp[hash(j+len1-mid,j+mid-1)];
	}
	printf("%lld\n",ans);
}