首页 > 代码库 > kmp
kmp
Given two sequences of numbers : a[1], a[2], ...... , a[N], and b[1], b[2], ...... , b[M] (1 <= M <= 10000, 1 <= N <= 1000000). Your task is to find a number K which make a[K] = b[1], a[K + 1] = b[2], ...... , a[K + M - 1] = b[M]. If there are more than one K exist, output the smallest one.
Input
The first line of input is a number T which indicate the number of cases. Each case contains three lines. The first line is two numbers N and M (1 <= M <= 10000, 1 <= N <= 1000000). The second line contains N integers which indicate a[1], a[2], ...... , a[N]. The third line contains M integers which indicate b[1], b[2], ...... , b[M]. All integers are in the range of [-1000000, 1000000].
Output
For each test case, you should output one line which only contain K described above. If no such K exists, output -1 instead.
Sample Input
213 51 2 1 2 3 1 2 3 1 3 2 1 21 2 3 1 313 51 2 1 2 3 1 2 3 1 3 2 1 21 2 3 2 1
Sample Output
6-1
- #include<stdio.h>
- #include<string.h>
- int next[10005];
- int s[1000005],p[10005];
- int n,m;
- void getNext(int p[])
- {
- int i=0,j=-1,len=m-1;
- next[0]=-1;
- while(i<len)
- {
- if(j==-1||p[i]==p[j])
- {
- i++,j++;
- next[i]=j;}
- else j=next[j];
- }
- }
- int kmp(int n,int m)
- {
- int i=0,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 t,i,j;
- scanf("%d",&t);
- while(t--)
- {
- scanf("%d%d",&n,&m);
- for(i=0;i<n;i++)
- scanf("%d",&s[i]);
- for(j=0;j<m;j++)
- scanf("%d",&p[j]);
- getNext(p);
- printf("%d\n",kmp(n,m));
- }
- return 0;
- }
#include<stdio.h>#include<string.h>int next[10005];int s[1000005],p[10005];int n,m;void getNext(int p[]){ int i=0,j=-1,len=m-1; next[0]=-1; while(i<len) { if(j==-1||p[i]==p[j]) { i++,j++; next[i]=j;} else j=next[j]; }}int kmp(int n,int m){ int i=0,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 t,i,j; scanf("%d",&t); while(t--) { scanf("%d%d",&n,&m); for(i=0;i<n;i++) scanf("%d",&s[i]); for(j=0;j<m;j++) scanf("%d",&p[j]); getNext(p); printf("%d\n",kmp(n,m)); } return 0;}
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。