HDU - 5918 小菜鸡的一点理解

最新推荐文章于 2020-10-04 21:18:01 发布

李自行h

最新推荐文章于 2020-10-04 21:18:01 发布

阅读量123

点赞数

分类专栏： ACM KMP 文章标签： c++

本文链接：https://blog.csdn.net/sssssss521/article/details/108610020

版权

ACM 同时被 2 个专栏收录

18 篇文章 1 订阅

订阅专栏

KMP

1 篇文章 0 订阅

订阅专栏

//小白，刚学的KMP在比赛时看出来是KMP模板题队友一发过了，在补题时一直过不去…
//在网上查了好久一直没查到两个KMP模板有什么不同最后借鉴大佬的博客过了
//题意：给两个串a,b,一个p，问在a中每次间隔p的情况下能分出来几个b（包含几个b）
//思路：KMP模板题只不过这次匹配a，b时每次要加上p其他的没什么不同

参考大佬的博客
链接: link.
https://blog.csdn.net/wjmwsgj/article/details/79994284

//AC代码
#include<iostream>
#include<sstream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<map>
#include<stack>
#include<queue>
using namespace std;
int t,n,m,p,q=0,ans=0;
int  a[1200000],b[1200000];
int next_p[1200000];
void getnext()//求NEXT数组
{
    memset(next_p,0,sizeof(next_p));
    int kk=-1,i=0;
    next_p[0]=-1;
    while(i<m)
    {
        if(kk==-1||b[i]==b[kk])
        {
            i++;
            kk++;
            next_p[i]=kk;
        }
          else kk=next_p[kk];
    }


}
void KMP()
{
    getnext();
    int kk=0,i=0;
    for(int j=0;j<p;j++)//范围小于p，若等于p的话为j=0的子串，大于p的话均为前面的子串
    {
        kk=0;
        i=j;
        while(i<n)
        {
            if(kk==-1||b[kk]==a[i])i+=p,kk++;//匹配时每次加p
               else kk=next_p[kk];
            if(kk==m)
            {
                ans++;
                kk=next_p[kk];
            }


        }

    }

}
int main()
{
ios::sync_with_stdio(0);//缩短时间，要不然会超时
    cin>>t;
    int g=0;
    while(t--)
    {ans=0;
        cin>>n>>m>>p;
        for(int i=0;i<n;i++)
          cin>>a[i];
        for(int j=0;j<m;j++)
          cin>>b[j];
         KMP();
        printf("Case #%d: %d\n",++g,ans);

    }
}

下面是另一个模板，不知道为什么过不去…

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<map>
#include<sstream>
#include<stack>
#include<queue>
using namespace std;
 int t,n,m,p,a[1200000],b[1200000],ans=0;
 int nextp[1200000];
void next_p()
{
    memset(nextp,0,sizeof(nextp));
    int kk=-1;
    nextp[0]=-1;
    for(int i=1;i<m;i++)
         {while(kk!=-1&&b[kk+1]!=b[i])
             kk=nextp[kk];
             if(b[kk+1]==b[i])  kk++;
             nextp[i]=kk;}
}
void kmp()
{
    next_p();
    int k=n,kk=-1,l=m;
    for(int i=0;i<p;i++)
      {
          for(int j=i;j<k;j+=p)
            {
                while(kk!=-1&&b[kk+1]!=a[j])
                   kk=nextp[kk];
                     if(b[kk+1]==a[j]) kk++;
                if(kk==l-1)
                {
                    ans++;
                    kk=nextp[kk];
                }

            }
      }
}
int main()
{
    //ios::sync_with_stdio(0);
int yy=0;
    scanf("%d",&t);
    while(t--)
    {ans=0;
        scanf("%d%d%d",&n,&m,&p);
        for(int i=0;i<n;i++)
          scanf("%d",&a[i]);
          for(int j=0;j<m;j++)
            scanf("%d",&b[j]);
              kmp();
        printf("Case #%d: %d\n",++yy,ans);
    }
}