数据结构实验之串三:KMP应用
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 next[1000000];
int s[1000000],p[1000000];
void getnext(int m)
{
int i=-1,j=0;
next[0]=-1;
while(j<m-1)
{
if(i==-1||p[i]==p[j])
{
++i;
++j;
next[j]=i;
}
else
{
i=next[i];
}
}
}
int kmp(int n,int m,int x)
{
int i=x,j=0;
while (i<n&&j<m)
{
if (j==-1||s[i]==p[j])
{
i++;
j++;
}else j=next[j];
}
if (j==m)
{
return i-j+1;
}else return -1;
}
int main()
{
int n,m,x,i;
scanf("%d",&n);
for (i=0;i<n;i++)
{
scanf("%d",&s[i]);
}
scanf("%d",&m);
for (i=0;i<m;i++)
{
scanf("%d",&p[i]);
}
getnext(m);
x=kmp(n,m,0);
if (x==-1)
{
printf("-1\n");
}else
{
if (kmp(n,m,x)==-1)
{
printf("%d %d\n",x,x+m-1);
}else printf("-1\n");
}
}
KMP算法。