首页 > 代码库 > [kmp]HDU1711 Number Sequence

[kmp]HDU1711 Number Sequence

题目大意

题意:给T组数据,每组有长度为n和m的母串和模式串。判断模式串是否是母串的子串,如果是输出最先匹配完成的位置,否则输出-1.

 

思考

直接套用模板。把char改成int。kmp函数中在模式串遍历到结尾的时候return,若没遍历到结尾,也就是不是子串返回-1

 

#include <iostream>  #include  <cstdio>  #include <cstring>  using namespace std;  int nexta[10005],a[1000005],s[10005];  int n,m;  void getfail(int* p,int* f){    f[0]=f[1]=0;    for(int i=1;i<m;i++)    {        int j=f[i];        while(j&&p[j]!=p[i]) j=f[j];        f[i+1]=(p[i]==p[j])?j+1:0;    }}int kmp(int* t,int* p,int* f){    getfail(p,f);    int j=0;    for(int i=0;i<n;i++)    {        while(j&&p[j]!=t[i]) j=f[j];        if(p[j]==t[i]) j++;        if(j==m)        {            return i-j+2;        }    }    return -1;} int main()  {      int T;      scanf("%d",&T);      while(T--)      {          scanf("%d%d",&n,&m);          for(int i = 0;i < n; i ++)          {              scanf("%d",&a[i]);          }          for(int j = 0; j < m;j ++)          {              scanf("%d",&s[j]);          }          printf("%d\n",kmp(a,s,nexta));      }      return 0;  }  

 

[kmp]HDU1711 Number Sequence