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
Output
2 4
#include <stdio.h>
#include <stdlib.h>
int a[1100000],b[1100000];
int next[1100000];
void getnext(int b[],int m)
{
int i=0,j=-1;
next[0]=-1;
while(i<m)
{
if(j==-1||b[i]==b[j])
{
i++;
j++;
if(b[i]!=b[j])next[i]=j;
else next[i]=next[j];
}
else j=next[j];
}
}
int kmp(int a[],int b[],int n,int m,int p)
{
int i=p,j=0;
getnext(b,m);
while(i<n&&j<m)
{
if(j==-1||a[i]==b[j])
{
i++;
j++;
}
else j=next[j];
}
if(j>=m)return i-j+1;
else return -1;
}
int main()
{
int i,n,m,x,y;
scanf("%d",&n);
for(i=0; i<n; i++)scanf("%d",&a[i]);
scanf("%d",&m);
for(i=0; i<m; i++)scanf("%d",&b[i]);
x=kmp(a,b,n,m,0);
if(x==-1)//没有可以匹配的直接输出-1
printf("-1\n");
else//找到了可以匹配的
{
y=kmp(a,b,n,m,x);//确定一下找到的可以匹配的是不是唯一的,要从刚才找到的可以匹配的下一个开始找,在KMP算法中,可以匹配的位置实际是从x-1开始的,只是为了方便输出结果(题目的第一个元素下标为1,而程序中第一个元素下标为0),将下标加了1。因此用 y=kmp(a,b,n,m,x),x正好是之前开始匹配位置的下一个
if(y==-1)printf("%d %d\n",x,x+m-1);//后面没有再能匹配的了,说明能找到唯一匹配的
else printf("-1\n");//否则,后面还有能匹配的,就输出-1
}
return 0;
}