HDOJ_1711_KMP 求匹配位置

/* HDOJ_1711_KMP 求匹配位置
*
* I really like this KMP in door
*
* Author : a_clay  2014/05/06
*/
#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <vector>
#include <algorithm>
#include <cmath>
#define Bug cout << "here\n";
using namespace std;

const int N = 1000005;
const int M = 10005;

int s
;
int t[M];
int next[M];

void get_next(int len) {
int i, j;
i = 0;
j = -1;
next[0] = -1;
while (i < len - 1) {
if (j == -1 || t[i] == t[j]) {
i++; j++; next[i] = j;
}
else {
j = next[j]; // 核心中的核心，KMP中的KMP。
}
}
}
int kmp(int sl, int tl) {
int i, j;
i = 0;
j = 0;
while (i < sl && j < tl) {
if (j == -1 || s[i] == t[j]) { //j == -1  是第一个就没有匹配上，类似于传统方法，i要向后移动，因为-1的巧妙设置，所以j依然很好的移动，i 还是从第一位 开始从 j==0比较
i++; j++;
}
else { j = next[j];}
}
if (j == tl) {
return i-j+1;
}
else return -1;
}

// abcabcababcabcabdef
// abcabcabd

int main() {
int T, n, m, ans;
cin >> T;
while (T--) {
cin >> n >> m;
int i;
for (i = 0; i < n; i++) {
scanf("%d", s+i);
}
for (i = 0; i < m; i++) {
scanf("%d", t+i);
}
get_next(m);
ans = kmp(n, m);
cout << ans << endl;
}
return 0;
}