[bzoj4084][Sdoi2015]双旋转字符串_hash

双旋转字符串 bzoj-4084 Sdoi-2015

题目大意：给定两个字符串集合 S 和 T 。其中 S 中的所有字符串长度都恰好为 N ，而 T 中所有字符串长度都恰好为 M 。且 N+M 恰好为偶数。如果记 S 中字符串全体为 S1，S2，...，STotalS ,而 T 中字符串全体为 T1，T2，...，TTotalT 。现在希望知道有多少对 <i,j> ，满足将 Si 和 Tj 拼接后得到的字符串 Si+Tj 满足双旋转性。一个长度为偶数字符串 W 可以表示成两段长度相同的字符串的拼接，即W=U+V。如果 V 可以通过 U 旋转得到，则称 W 是满足双旋转性的。比如说字符串 U=“vijos”可以通过旋转得到“ijosv”，“josvi”，“osvij” 或“svijo”。那么“vijosjosvi”就是满足双旋转性的字符串。

想法：我们将小的集合所有串hash然后存起来，这一步是容易的。对于大串我们将他所有旋转后的前(len1-mid)也存起来，这里只需要将每一个大串在尾部copy一遍，然后hash前缀和即可。mid的意思就是大、小串的长度平均值。然后暴力枚举匹配即可。

最后，附上丑陋的代码... ...

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <map>
#define ll long long
#define ull unsigned long long
#define N 4000005
#define base 233
using namespace std;
int n,m,len1,len2;
ull p
,f
;
ll ans;
string s
,t
;
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;
for(int i=1;i<=len1+len2;i++) p[i]=p[i-1]*base;
int mid=(len1+len2)>>1;
if (len1<len2)
{
for(int i=1;i<=n;i++) cin >> t[i] ;
for(int i=1;i<=n;i++) cin >> s[i] ;
swap(n,m);swap(len1,len2);
}
else
{
for(int i=1;i<=n;i++) cin >> s[i] ;
for(int i=1;i<=n;i++) cin >> t[i] ;
}
for(int i=1;i<=n;i++)
{
ull h=0;
for(int j=0;j<=len2-1;j++) h=h*base+t[i][j]-'a'+1;
mp[h]++;
}
for(int i=1;i<=n;i++)
{
for(int j=0;j<=mid*2-1;j++) f[j+1]=f[j]*base+s[i][j%mid]-'a'+1;
ull h=0;
for(int j=mid;j<=len1-1;j++) h=h*base+s[i][j]-'a'+1;
for(int j=1;j<=mid;j++) if (hash(j,j+len1-mid-1)==h) ans+=mp[hash(j+len1-mid,j+mid-1)];
}
printf("%lld\n",ans);
}

小结：map真好用...hash真强... ...