A - Number Sequence HDU1005 ( kmp 算法+整数数组的应用)

标签：

-1

#include<cstdio>
#include<algorithm>
#include<cmath>
//#include<bits/stdc++.h>
using namespace std;
template<class T>inline T read(T&x)
{
    char c;
    while((c=getchar())<=32)if(c==EOF)return 0;
    bool ok=false;
    if(c=='-')ok=true,c=getchar();
    for(x=0; c>32; c=getchar())
        x=x*10+c-'0';
    if(ok)x=-x;
    return 1;
}
template<class T> inline T read_(T&x,T&y)
{
    return read(x)&&read(y);
}
template<class T> inline T read__(T&x,T&y,T&z)
{
    return read(x)&&read(y)&&read(z);
}
template<class T> inline void write(T x)
{
    if(x<0)putchar('-'),x=-x;
    if(x<10)putchar(x+'0');
    else write(x/10),putchar(x%10+'0');
}
template<class T>inline void writeln(T x)
{
    write(x);
    putchar('\n');
}
//-------ZCC IO template------
const int maxn=1000001;
const double inf=999999999;
#define lson (rt<<1),L,M
#define rson (rt<<1|1),M+1,R
#define M ((L+R)>>1)
#define For(i,t,n) for(int i=(t);i<(n);i++)
typedef long long  LL;
typedef double DB;
typedef pair<int,int> P;
#define bug printf("---\n");
#define mod  1000000007

int nex[maxn];

void getnext(int*a,int lena)
{
    int i=0,j=-1;
    nex[0]=-1;
    while(i<lena)
    {
        if(j==-1||a[i]==a[j])
            nex[++i]=++j;
        else
            j=nex[j];
    }
}

int kmp(int *a,int lena,int *b,int lenb)
{
    int i=0,j=0;
    while(i<lena&&j<lenb)
    {
        if(j==-1||a[i]==b[j])
            i++,j++;
        else
            j=nex[j];
        if(j==lenb)return i-j+1;
    }
    return -1;
}
int a[maxn],b[maxn];


int main()
{
    //#ifndef ONLINE_JUDGE
    //freopen("in.txt","r",stdin);
    //freopen("zccccc.txt","w",stdout);
    //#endif // ONLINE_JUDGE
    int n,m,i,j,t,k;
    int T;
    read(T);
    while(T--)
    {
        read_(n,m);
        For(i,0,n)read(a[i]);
        For(i,0,m)read(b[i]);
        getnext(b,m);
        writeln(kmp(a,n,b,m));
    }
    return 0;
}

A - Number Sequence HDU1005 ( kmp 算法+整数数组的应用)

标签：

原文地址：http://blog.csdn.net/u013167299/article/details/45078439