数据结构实验之串三：KMP应用

数据结构实验之串三：KMP应用
Time Limit: 1000 ms Memory Limit: 65536 KiB

Problem Description
有n个小朋友，每个小朋友手里有一些糖块，现在这些小朋友排成一排，编号是由1到n。现在给出m个数，能不能唯一的确定一对值l和r(l <= r)，使得这m个数刚好是第l个小朋友到第r个小朋友手里的糖块数？

Input
首先输入一个整数n，代表有n个小朋友。下一行输入n个数，分别代表每个小朋友手里糖的数量。

之后再输入一个整数m，代表下面有m个数。下一行输入这m个数。

Output
如果能唯一的确定一对l,r的值，那么输出这两个值，否则输出-1

Sample Input
5
1 2 3 4 5
3
2 3 4
Sample Output
2 4
本题旨在考察对KMP的应用。KMP简而言之就是建立一个next数组将子串的最大前后缀值存起来，
然后在调用。

#include <bits/stdc++.h>
#include <iostream>

using namespace std;

int a[1000010],b[1000010],nex[1000010];
int n,m;//定义为全局变量
void Find_next(int b[])
{
    int j,k;
    k=-1;
    j=0;
    nex[0]=-1;
    while(j<m)
    {
        if(k==-1||b[j]==b[k])//k==-1很关键，注意体会
        {
            k++;
            j++;
            nex[j]=k;
        }
        else
        {
            k=nex[k];
        }
    }
}
int KMP(int a[],int b[],int z)//注意z的传值
{
    Find_next(b);
    int i,j;
    i=z;
    j=0;
    while(i<n&&j<m)
    {
        if(j==-1||a[i]==b[j])
        {
            i++;
            j++;
        }
        else j=nex[j];
    }
    if(j>=m)return i;///不能为i-m+1;因为当m==0时，返回的为1
    return -1;
}
int main()
{
    int i,t,k;
    for(i=0;i<n+1;i++)
    {
        a[i]=b[i]=nex[i]=0;
    }
    cin>>n;
    for(i=0;i<n;i++)
    {
        cin>>a[i];
    }
    cin>>m;
    for(i=0;i<m;i++)
    {
        cin>>b[i];
    }
    t=KMP(a,b,0);
    if(t==-1)cout<<t<<endl;
    else
    {
        k=KMP(a,b,t-m+1);//注意此处调用开始的值
        if(k==-1)cout<<t-m+1<<" "<<t<<endl;
        //else if(k==-1&&m==1)cout<<t<<endl;
        else cout<<"-1"<<endl;
    }
    return 0;
}

/***************************************************
Result: Accepted
Take time: 352ms
Take Memory: 1356KB
****************************************************/