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

数据结构实验之串三：KMP应用
Time Limit: 1000MS Memory Limit: 65536KB
Problem Description

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

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

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

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

5
1 2 3 4 5
3
2 3 4

Example Output

2 4

#include <stdio.h>
int a[1000000],b[1000000],next[1000000];
void knext(int len);
int kmp(int len1,int len2,int q);
int main(){
    int n,m;
    scanf("%d",&n);
    for (int i=0;i<n;i++){
        scanf("%d",&a[i]);
    }
    scanf("%d",&m);
    for (int i=0;i<m;i++){
        scanf("%d",&b[i]);
    }
    knext(m);
    int aa=kmp(n,m,0);
    if (aa!=-1){
        int bb=kmp(n,m,aa+m-1);
        if (bb==-1){
            printf("%d %d\n",aa,aa+m-1);
        }
        else{
            printf("-1\n");
        }
    }
    else{
        printf("-1\n");
    }
    return 0;
}
void knext(int len){
    int i=0,j=-1;
    next[0]=-1;
    while(i<len){
        if (j==-1||b[i]==b[j]){
            i++;
            j++;
            next[i]=j;
        }
        else{
            j=next[j];
        }
    }
}
int kmp(int len1,int len2,int q){
    int i=q,j=0;
    while(i<len1&&j<len2){
        if (j==-1||a[i]==b[j]){
            i++;
            j++;
        }
        else{
            j=next[j];
        }
    }
    if (j>=len2){
        return i-len2+1;
    }
    else{
        return -1;
    }
}