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
#include <stdio.h>
#include <stdlib.h>
int n,m;
int a[1000001],b[1000001];
int next[1000001];
void get_next(int str[])
{
int i=0,j=-1;
next[0]=-1;
while(i<=m-1)
{
if(j==-1||str[i]==str[j])
{
i++;
j++;
next[i]=j;
}
else j=next[j];
}
}
int KMP(int x)
{
int i=x,j=0;
while(i<=n-1&&j<=m-1)
{
if(j==-1||a[i]==b[j])
{
i++;
j++;
}
else j=next[j];
}
if(j==m) return (i-(m-1));
else return -1;
}
int main()
{
scanf("%d",&n);
int i;
for(i=0; i<=n-1; i++) scanf("%d",&a[i]);
scanf("%d",&m);
for(i=0; i<=m-1; i++) scanf("%d",&b[i]);
get_next(b);
int k;
k=KMP(0);
if(k!=-1)
{
int c;
c=KMP(k);
if(c==-1) printf("%d %d\n",k,k+(m-1));
else printf("%d\n",-1);
}
else printf("%d\n",-1);
return 0;
}