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

数据结构实验之串一：KMP简单应用
Time Limit: 1000 ms Memory Limit: 65536 KiB

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
cjx

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

char str1[1000005],str2[1000005];
int nex[1000005];
void get_next();
int kmp();
int main()
{
    while(scanf("%s%s",str1,str2)!=EOF)
    {
        get_next();
        printf("%d\n",kmp());
    }
    return 0;
}
void get_next()
{
    int len2,i,j;
    len2=strlen(str2);
    i=0;
    j=0;
    nex[0]=0; //next数组第一位初始化为0
    for(i=1;i<len2;i++)
    {
        j=nex[i-1]; //与上一位next数组下标对应的字符比较
        while(j!=0&&str2[i]!=str2[j])
        {
            j=nex[j-1]; //若j=0或不匹配，直到找到匹配字符的下标为止
        }
        if(str2[i]==str2[j])
        {
            nex[i]=j+1; //匹配
        }
        else
        {
            nex[i]=0; //不匹配
        }
    }
}
int kmp()
{
    int i,j,len1,len2;
    len1=strlen(str1);
    len2=strlen(str2);
    i=0; //对应主串
    j=0; //对应模式串
    while(i<len1&&j<len2)
    {
        if(str1[i]==str2[j]) //匹配
        {
            i++;
            j++;
        }
        else
        {
            if(j==0) //失配且j=0，由于next[0-1]无对应数值，故要加这个判断条件
                i++;
            else
                j=nex[j-1]; //注意nex下标为j-1，而不是j
        }
    }
    if(j==len2) //模式串全部匹配
        return i-len2+1;
    else
        return -1;
}