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 <string.h>
int i,j,n,m;
int t[1000010],p[1000010];
int next[1000010];
void getnext(int p[])
{
i=0,j=-1;
next[0]=-1;
while(i<m)
{
if(j==-1||p[i]==p[j])
{
i++;
j++;
next[i]=j;
}
else j=next[j];
}
}
int KMP(int t[],int p[],int x)
{
getnext(p);
i=x,j=0;
while(i<n&&j<m)
{
if(t[i]==p[j]||j==-1)
{
i++;
j++;
}
else j=next[j];
}
if(j>=m)
{
return i-m+1;
}
else return -1;
}
int main()
{
int k;
scanf("%d",&n);
for(k=0; k<n; k++)
{
scanf("%d",&t[k]);
}
scanf("%d",&m);
for(k=0; k<m; k++)
{
scanf("%d",&p[k]);
}
int flag=KMP(t,p,0);
if(flag!=-1)
{
int k=KMP(t,p,i-m+1);
if(k==-1)
{
printf("%d %d\n",flag,flag+m-1);
}
else printf("-1\n");
}
else printf("-1\n");
return 0;
}