数据结构实验之串三: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
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
int next[10000002]= {0};
int a1[10000002]= {0},a2[10000002]= {0};
int n,m;
int s=0;
void get(int a[])
{
int i=0,t=-1 ;
next[i]=t;
while(i<m)
{
if(t==-1||a[i]==a[t])
{
t++;
i++;
next[i]=t;
}
else t=next[t];
}
}
void kmp(int a1[],int a2[])
{
get(a2);
int i,j;
i=0;
j=0;
int a,b;
while(i<n)
{
if(j==-1||a1[i]==a2[j])
{
j++;
i++;
}
else j=next[j];
if(j==m) //记录含有那个子串的次数
{
s++;
if(s==1)
{
a=i,b=j;
}
if(s>1)break; //大于1就结束
j=0;
i=i-m+1; //必须回到那一段的开头的下一个
}
}
if(s==1)printf("%d %d",a-b+1,a);
else printf("-1");
}
int main()
{
scanf("%d",&n);
int i;
for(i=0; i<n; i++)
{
scanf("%d",&a1[i]);
}
scanf("%d",&m);
for(i=0; i<m; i++)
scanf("%d",&a2[i]);
kmp(a1,a2);
return 0;
}