HDUOJ1711Number Sequence（KMP）

链接：HDUOJ1711Number Sequence
题意：
输入案例数t，
每组案例给定两组数，a[1]~a[N]andb[1]~b[M]
(1 <= M <= 10000, 1 <= N <= 1000000)
找出b在a中第一次出现的位置，若没有输出-1
思路：KMP循环判断
代码：

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
#define N 1000010
int ne[N];
int a[N];
int b[N];
int t,n,m;
int re;
void getnext(int* s){
    //int len=sizeof(s);
    int x=0,y=-1;
    ne[0]=-1;
    while(x<m){
        if((y==-1)||(s[x]==s[y]))
            ne[++x]=++y;
        else y=ne[y];
    }
}//求ne[]
void kmp(int a[],int b[])
{
    getnext(b);
    /*for(int i=0;i<m;i++)
        cout<<ne[i]<<' ';*/
        //上面两行就输出一下ne[]看看来着
        int i=0,j=0;
        int no=0;
        while((i<n)&&(j<m))
        {
            if((j==-1)||(a[i]==b[j]))
            {
                i++;
                j++;
            }
                else j=ne[j];
            if(j==m)
            {
                no=1;
                re=i-m+1;
                //break;
            }
            //cout<<i<<endl;
        }
        if(no==1)
        printf("%d\n",re);
        else //cout<<-1<<endl;
            printf("-1\n");

}
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++)
            scanf("%d",&a[i]);
        for(int i=0;i<m;i++)
            scanf("%d",&b[i]);
        kmp(a,b);
    }
    return 0;
}

完了以后我发现一件神奇的事emmmm就是本来直接写在main函数里的代码，把它写成函数可以缩短时间……

---

现在是2018/8/9
如果有人看到这里的话
那祝你幸福[笔芯]
其实就是
我又重新写了一遍代码
希望有人能夸我的coding style
就酱，啾咪~

#include<cstdio>
#include<iostream>
#include<set>
#include<cstring>

using namespace std;

int n, m;
int a[1000000 + 10];
int b[10000 + 10];
int ne[10000 + 10];
void getnext(int s[])
{
    int i = 0, j = -1;
    ne[0] = -1;
    while(i < m)
    {
        if(j == -1 || s[i] == s[j])
        {
            ne[++i] = ++j;
        }
        else
        {
            j = ne[j];
        }
    }
}
int kmp(int a[], int b[])
{
    getnext(b);
    int ans = -1;
    int i = 0, j = 0;
    while((i < n) && (j < m))
    {
        if((j == -1) ||(a[i] == b[j]))
        {
            i++;
            j++;
        }
        else j = ne[j];
        if(j == m)
        {
            ans = i -m + 1;
        }
    }
    return ans;
}
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 i = 0; i < m; i++)
            scanf("%d", &b[i]);
        int ans = kmp(a, b);
        printf("%d\n",ans);
    }
    return 0;
}