TZOJ 5159: Yuchang ans Zixiang’s Magic
## 题目地址: http://www.tzcoder.cn/acmhome/problemdetail.do?method=showdetail&id=5159
题目描述:
需要上面数字和下面数字每一个拼起来一样。
例如样例:
解题:
解法一;
使用kmp算法:
先预处理他们的两两之间的差值:
发现模式串和匹配串如果互为相反数就可以匹配。
代码:
#include<bits/stdc++.h>
using namespace std;
int n,m;
const int N=10005,M=1000005;
int a[N],b[M],p[N],s[M],nex[N];
int main()
{
while(~scanf("%d %d",&n,&m))
{
memset(nex,0,sizeof nex);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<n;i++) p[i]=a[i+1]-a[i];
for(int i=1;i<=m;i++)scanf("%d",&b[i]);
for(int i=1;i<m;i++) s[i]=b[i+1]-b[i];
if(n==1)//注意当n==1时需要特判。
{
printf("%d\n",m);
continue;
}
for(int i=2,j=0;i<n;i++)
{
while(j&&p[i]!=p[j+1])j=nex[j];
if(p[i]==p[j+1])j++;
nex[i]=j;
}
int cnt=0;
for(int j=0, i=1; i<m;i++)
{
while(j && p[j+1]+s[i]!=0) j = nex[j];
if(p[j+1]+s[i]==0)j++;
//cout<<i<<" "<<j<<" "<<c[j]<<" "<<d[i]<<endl;
if(j==n-1)
{
cnt++;
j=nex[j];
}
}
printf("%d\n",cnt);
}
}
解法二:
使用字符串哈希:
飞哥写的哈。凡是用kmp写的都可用字符串哈希乱搞搞出来。
先放代码:
#include<iostream>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
const int N=1e4+5,M=1e6+5,P=131;
typedef unsigned long long ull;
int n,m,w[12];
ull h[M],p[M]={1},mp;
ull pan(ull s,ull x)
{
x+=2e9;
for(int i=0;i<10;i++)
{
w[i]=x%10;
x/=10;
}
for(int i=9;i>=0;i--)
{
s=s*P+w[i]+1;
}
return s;
}
ull get(int l,int r)
{
return h[r]-h[l-1]*p[r-l+1];
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
mp=0;
int ans=0,a,b;
scanf("%d",&a);
for(int i=1;i<n;i++)
{
scanf("%d",&b);
mp=pan(mp,a-b);a=b;
}
scanf("%d",&a);
for(int i=1;i<m;i++)
{
p[i]=p[i-1];
for(int j=0;j<10;j++)p[i]=p[i]*P;
scanf("%d",&b);
h[i]=pan(h[i-1],b-a);a=b;
if(i>=n-1)
{
if(get(i-n+2,i)==mp)ans++;
}
}
if(n==1)ans=m;
printf("%d\n",ans);
}
return 0;
}
虽说时间复杂度不如kmp但也时极佳的方法!
飞哥yyds!