题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1711
KMP的模板题, 很简单; 用来做模板。
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxn = 1e6 + 100;
int a[maxn], b[maxn];
int Next[maxn];
void get_Next(int m)
{
Next[0] = -1;
for(int i = 0; i < m; i++)
{
int j = Next[i];
while(j!=-1 && b[i]!=b[j])
j = Next[j];
Next[i+1] = j + 1;
}
}
void solve(int n, int m)
{
int j = 0, ans = -1;
for(int i = 0; i < n; i++)
{
while(j!=-1 && a[i]!=b[j])
j = Next[j];
j = j + 1;
if(j == m)
{
ans = i - m + 2;
break;
}
}
printf("%d\n", ans);
}
int main()
{
int T, n, m;
scanf("%d", &T);
while(T--)
{
scanf("%d%d", &n, &m);
for(int i = 0; i < n; i++)
scanf("%d", &a[i]);
for(int j = 0; j < m; j++)
scanf("%d", &b[j]);
get_Next(m);
solve(n, m);
}
return 0;
}