TZOJ 5159 类kmp或字符串哈希

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！