数据结构实验之串三:KMP应用
Time Limit: 1000 ms Memory Limit: 65536 KiB
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
Hint
Source
windream
#include <stdio.h>
#include <string.h>
int next[10000005];
int a[10000005];
int b[10000005];
void set(int b[],int next[],int m)
{
int i=0,i1=-1;
next[i]=i1;
while(i<m)
{
if(i1==-1||b[i]==b[i1])
{
i++;
i1++;
next[i]=i1;
}
else i1=next[i1];
}
}
int kmp(int a[],int b[],int n,int m)
{
set(b,next,m);
int i=0,i1=0;
while(i<n&&i1<m)
{
if(i1==-1||a[i]==b[i1])
{
i++;
i1++;
}
else i1=next[i1];
}
if(i1==m)
return i-m+1;
else return -1;
}
int main()
{
int n,m;
scanf("%d",&n);
int i;
for(i=0; i<n; i++)
scanf("%d",&a[i]);
scanf("%d",&m);
for(i=0; i<m; i++)
scanf("%d",&b[i]);
kmp(a,b,n,m);
int k;
k=kmp(a,b,n,m);
if(k!=-1)
{
int t=kmp(a+k,b,n,m);
if(t==-1)printf("%d %d\n",k,k+m-1);
else printf("-1\n");
}
else printf("-1\n");
return 0;
}