hdu 1711 Number Sequence (KMP - 初步)

原创于 2012-09-06 11:51:11 发布 · 395 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#include

icpc总论专栏收录该内容

68 篇文章

订阅专栏

本文介绍了一种高效的字符串匹配算法——KMP算法，并通过C++代码详细展示了该算法的具体实现过程。包括如何构建next数组以加快匹配速度，以及如何进行主串与模式串之间的比较。

#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>

using namespace std;
const int N = 1000005;
const int M = 10005;

int s[N];
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];
        }
    }
}
int kmp(int sl, int tl) {
    int i = 0, j = 0;
    while(i < sl && j < tl) {
        if(j == -1 || s[i] == t[j]) {
            i++, j++;
        }
        else {
            j = next[j];
        }
    }
    if(j == tl) {
        return i - j + 1;
    }
    else return -1;
}
int main() {
    int T, n, m, ans;
    cin >> T;
    while(T--) {
        scanf("%d%d", &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);
        printf("%d\n", ans);
    }
    return 0;
}