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

Problem Description

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

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

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

abc
a
123456
45
abc
ddd
Example Output

1
4
-1

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define max 1001111
int next[max];
char s1[max], s2[max];
int n, m;
void get_next()
{
    int i, j;
    i = 1; j=0;
    next[1] = 0;
    while(i<m)
    {
        if(j==0||s2[i]==s2[j])
        {
            i++;j++;
            next[i] = j;
        }
        else
        {
            j = next[j];
        }
    }
}
void kmp()
{
    int i, j;
    i = 1;
    j = 1;
    while(i<n&&j<m)
    {
        if(j==0||s1[i]==s2[j])
        {
            i++;j++;
        }
        else
        {
            j = next[j];
        }
    }
    if(j>=m)
    {
        printf("%d\n", i-m+1);
    }
    else
        printf("-1\n");
}
int main()
{
    while(gets(s1))
    {
        gets(s2);
        n = strlen(s1);
        m = strlen(s2);
        get_next();
        kmp();
    }
}