数据结构实验之串三: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
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
int a[1000100],b[1000100],nex[1000100];
void next(int b[],int m)
{
nex[0]=-1;
int k=-1;
for(int i=1;i<m;i++)
{
while(k>-1&&b[k+1]!=b[i])
{
k=nex[k];
}
if(b[k+1]==b[i])
k=k+1;
nex[i]=k;
}
}
int kmp(int a[],int b[],int n,int m)
{
next(b,m);
int k=-1;
int f;
int sum=0;
for(int i=0;i<n;i++)
{
while(k>-1&&b[k+1]!=a[i])
{
k=nex[k];
}
if(b[k+1]==a[i])
{
k=k+1;
}
if(k==m-1)
{
sum++;
if(sum>=2)return -1;
f=i-k+1;//不能是i-m+1,因为这样就会少一位;
i=i-m+1;//i是一个个的向后比较的,外循环有个i++,这样可以直接比较下一个,f是记录下来第一个位置,i与k都要重新赋值
k=-1;//k重新赋值;
}
}
if(sum==0)return -1;
else return f;
}
int main()
{
int n,m;
scanf("%d",&n);
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
}
scanf("%d",&m);
for(int i=0;i<m;i++)
{
scanf("%d",&b[i]);
}
int f=kmp(a,b,n,m);
if(f==-1)printf("-1\n");
else printf("%d %d\n",f,f+m-1);
return 0;
}