hdu 1711 Number Sequence 解题报告

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1711

题目意思：给出一条有n个数的序列a[1]，a[2]，......，a[n]，和一条有m 个数的序列b[1]，b[2]，......，b[m]，求出b[1]，b[2]，...，b[m]在序列a中完全匹配时，在序列a中的位置，如果找不到输出-1.

        这几天一直在学kmp，该题算是kmp的入门题吧。有个地方要稍稍注意，代码中，主串和模式串的比较初始值为-1，-1，否则如果从0开始，会默认第一个字符是相等的！！！

      

#include <iostream>
#include <cstdlib>
#include <cstdio>
using namespace std;

const int N = 1e6 + 5;
const int M = 1e4 + 5;
int next[M], n, m;
int a[N], b[M];

void get_next(int *next)
{
    int i = 0;
    int j = -1;
    next[0] = -1;
    while (i < m)
    {
        if (j == -1 || b[i] == b[j])
        {
            i++;
            j++;
            next[i] = j;
        }
        else
            j = next[j];
    }
}

int main()
{
    int T, i, j;
    while (scanf("%d", &T) != EOF)
    {
        while (T--)
        {
            scanf("%d%d", &n, &m);
            for (i = 0; i < n; i++)
                scanf("%d", &a[i]);
            for (i = 0; i < m; i++)
                scanf("%d", &b[i]);
            get_next(next);
            i = -1, j = -1;    
            int ans = -1;
            while (i <= n && j <= m)
            {
                if (j == -1 || a[i] == b[j])    // 关键！
                {
                    ++i;
                    ++j;
                }
                else
                    j = next[j];
                if (j == m)
                {
                   ans = i - m + 1;
                   break;
                }
            }
            printf("%d\n", ans);
        }
    }
    return 0;
}