数据结构实验之串三：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

Code realization

#include <stdio.h>
#include <string.h>
int next[1000001],n,m;
int i,j;
void Get_next(int str2[])
{
    i = 0,j = -1;
    next[i] = j;
    while(i<m)
    {
        if(j==-1||str2[i]==str2[j])
        {
            i++;
            j++;
            next[i] = j;
        }
        else
            j = next[j];
    }
}
int KMP(int pos,int str1[],int str2[])
{
    Get_next(str2);
    i = pos, j = 0;
    while(i<n&&j<m)
    {
        if(j==-1||str1[i]==str2[j])
        {
            i++;
            j++;
        }
        else
            j = next[j];
    }
    if(j>=m)
        return i-m+1;
    else
        return 0;
}
int main()
{
    int str1[1000000], str2[1000000];
    int k;
    scanf("%d",&n);
    for(k=0;k<n;k++)
    {
        scanf("%d",&str1[k]);
    }
    scanf("%d",&m);
    for(k=0;k<m;k++)
    {
        scanf("%d",&str2[k]);
    }
    int flag = KMP(0,str1,str2);
    if(flag)
    {
        if(!KMP(i-m+1,str1,str2))
            printf("%d %d\n",flag,flag+m-1);
        else
            printf("-1\n");
    }
    else
        printf("-1\n");
    return 0;
}