HDU 1711 Number Sequence KMP

题目地址： http://acm.hdu.edu.cn/showproblem.php?pid=1711

AC代码：

 

#include <iostream>

#include <cstdio>

#include <cstdlib>

#include <cmath>

#include <cstring>

#include <string>

#include <vector>

#include <list>

#include <deque>

#include <queue>

#include <iterator>

#include <stack>

#include <map>

#include <set>

#include <algorithm>

#include <cctype>

using namespace std;



typedef long long LL;

const int N=10005;

const LL II=100000000;

const int INF=0x3f3f3f3f;

const double PI=acos(-1.0);



int next[N],tlen,wlen;

int text[1000009],word[N];



void getnext(int *p)

{

    int j=0,k=-1;

    next[0]=-1;

    while(j<wlen)//len是p的长度

    {

        if(k==-1||p[j]==p[k])

        {

            j++;    k++;

            next[j]=k;

        }

        else

            k=next[k];

    }

}



int kmp(int *text,int *word)//主串、模式串

{//返回从第几个开始出现模式串

    int i=0,j=0,k=0;

    while(i<tlen)

    {

        if(j==-1||text[i]==word[j])

        {

            i++;j++;

        }

        else

            j=next[j];

        if(j==wlen) return i-wlen+1;

    }

    return -1;

}



int main()

{

    int i,j,T;

    cin>>T;

    while(T--)

    {

        scanf("%d%d",&tlen,&wlen);

        for(i=0;i<tlen;i++)

            scanf("%d",&text[i]);

        for(i=0;i<wlen;i++)

            scanf("%d",&word[i]);

        getnext(word);

        int xx=kmp(text,word);

        printf("%d\n",xx);

    }

    return 0;

}