HDU 5782 Cycle(KMP+哈希)

时间:2025-01-26 23:33:26

http://acm.split.hdu.edu.cn/showproblem.php?pid=5782

题意:
给出两个长度相等的字符串,输出两个字符的每个前缀是否循环相同。

思路:

如果连个串循环相同的话,那一定可以找到一个位置,使得第一个串的前缀等于另一个串的后缀。这样的话其实就是扩展kmp的思想,kmp处理,然后用哈希来比较两段字符是否相等。

 #include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<vector>
#include<stack>
#include<queue>
#include<cmath>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
const int INF = 0x3f3f3f3f;
const int maxn = + ; const int seed =; int n;
int ans[maxn];
int f[maxn];
ll base[maxn];
ll Hash[][maxn];
char s1[maxn],s2[maxn]; int check(int flag, int l, int r)
{
if(l-==r) return ;
int L=r-l+;
ll tmp1 = Hash[flag][l]-Hash[flag][l+L]*base[L];
ll tmp2 = Hash[!flag][]-Hash[!flag][+L]*base[L];
return tmp1==tmp2;
} void kmp(char* T, char* P, int flag)
{
f[]=; f[]=;
for(int i=;i<n;i++)
{
int j=f[i];
while(j && P[i]!=P[j]) j=f[j];
f[i+]= P[i]==P[j]?j+:;
} int j=;
for(int i=;i<n;i++)
{
while(j && P[j]!=T[i]) j=f[j];
if(P[j]==T[i])
{
j++;
if(!ans[i]) ans[i]=check(flag,j,i);
}
}
} int main()
{
//freopen("in.txt","r",stdin);
base[]=;
for(int i=;i<=maxn;i++) base[i]=base[i-]*seed; while(~scanf("%s%s",s1,s2))
{
memset(ans,,sizeof(ans));
n=strlen(s1);
Hash[][n]=Hash[][n]=;
for(int i=n-;i>=;i--)
{
Hash[][i]=Hash[][i+]*seed+(s1[i]-'a'+);
Hash[][i]=Hash[][i+]*seed+(s2[i]-'a'+);
}
kmp(s1,s2,);
kmp(s2,s1,);
for(int i=;i<n;i++) printf("%d",ans[i]);
puts("");
}
return ;
}