数据结构实验之串一：KMP简单应用

Problem Description

给定两个字符串string1和string2，判断string2是否为string1的子串。

Input

输入包含多组数据，每组测试数据包含两行，第一行代表string1(长度小于1000000)，第二行代表string2（长度小于1000000），string1和string2中保证不出现空格。

Output

对于每组输入数据，若string2是string1的子串，则输出string2在string1中的位置，若不是，输出-1。

Sample Input

abc
a
123456
45
abc
ddd

Sample Output

1
4
-1

Hint

Source

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

int next[1000001];

void GetNextarr(char *p)  // 计算next数组
{
    int len = strlen(p);
    next[0] = -1;  // 初始化next[0]为-1
    int k = -1; 
    int j = 0;
    while(j < len - 1)
    {
        // p[k]表示前缀，p[j]表示后缀
        if (k == -1 || p[j] == p[k])
        {
            k++;
            j++;
            next[j] = k;  // 为next数组赋值
        }
        else
        {
            k = next[k];  // 前缀标识回溯
        }
    }
}

int KmpSearch(char *s, char *p)
{
    int i = 0;
    int j = 0;
    int len1 = strlen(s);
    int len2 = strlen(p);
    while(i < len1 && j < len2)
    {
        if (j == -1 || s[i] == p[j])  // 若匹配成功，或j=-1
        {
            i++;
            j++;
        }
        else
        {
            j = next[j];  // 模式串右移
        }
    }
    if (j == len2)
    {
        return i - j + 1;
    }
    else
    {
        return -1;
    }
}

int main()
{
    char str1[1000001];
    char str2[1000001];
    while(~scanf("%s %s", str1, str2))
    {
        GetNextarr(str1);
        printf("%d
", KmpSearch(str1, str2));
    }
    return 0;
}