数据结构实验之串三:KMP应用
Time Limit: 1000MS
Memory Limit: 65536KB
Problem Description
有n个小朋友,每个小朋友手里有一些糖块,现在这些小朋友排成一排,编号是由1到n。现在给出m个数,能不能唯一的确定一对值l和r(l <= r),使得这m个数刚好是第l个小朋友到第r个小朋友手里的糖块数?
Input
首先输入一个整数n,代表有n个小朋友。下一行输入n个数,分别代表每个小朋友手里糖的数量。
之后再输入一个整数m,代表下面有m个数。下一行输入这m个数。
Output
如果能唯一的确定一对l,r的值,那么输出这两个值,否则输出-1
Example Input
5 1 2 3 4 5 3 2 3 4
Example Output
2 4
Code realization
#include <stdio.h> #include <string.h> int next[1000001],n,m; int i,j; void Get_next(int str2[]) { i = 0,j = -1; next[i] = j; while(i<m) { if(j==-1||str2[i]==str2[j]) { i++; j++; next[i] = j; } else j = next[j]; } } int KMP(int pos,int str1[],int str2[]) { Get_next(str2); i = pos, j = 0; while(i<n&&j<m) { if(j==-1||str1[i]==str2[j]) { i++; j++; } else j = next[j]; } if(j>=m) return i-m+1; else return 0; } int main() { int str1[1000000], str2[1000000]; int k; scanf("%d",&n); for(k=0;k<n;k++) { scanf("%d",&str1[k]); } scanf("%d",&m); for(k=0;k<m;k++) { scanf("%d",&str2[k]); } int flag = KMP(0,str1,str2); if(flag) { if(!KMP(i-m+1,str1,str2)) printf("%d %d\n",flag,flag+m-1); else printf("-1\n"); } else printf("-1\n"); return 0; }