C - 数据结构实验之串二：字符串匹配

Description

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

Input

输入包含多组数据，每组测试数据包含两行，第一行代表string1，第二行代表string2，string1和string2中保证不出现空格。(string1和string2大小不超过100字符)

Output

对于每组输入数据，若string2是string1的子串，则输出"YES"，否则输出"NO"。

Sample

Input
abc
a
123456
45
abc
ddd

Output
YES
YES
NO

#include <iostream>
#include <bits/stdc++.h>

using namespace std;

void get_next(char t[],int next[])
{
    int i = 0;
    int j = -1;
    next[0] = -1;
    while(t[i] != '\0')
    {
        if(j == -1 || t[j] == t[i])
        {
            i++;
            j++;
            next[i] = j;
        }
        else
            j = next[j];
    }
}

bool kmp(char s[],char t[])
{
    int slen = strlen(s);
    int tlen = strlen(t);
    int next[tlen];
    int i = 0, j = 0;
    get_next(t,next);
    while(i<slen && j<tlen)
    {
        if(j == -1 || s[i] == t[j])
        {
            j++;
            i++;
        }
        else
            j = next[j];
    }
    if(j >= tlen)
        return true;
    else
        return false;
}

int main()
{
    ios::sync_with_stdio(false);

    char s[1000005],t[1000005];
    while(cin>>s>>t)
        if(kmp(s,t))
            cout<<"YES\n";
        else
            cout<<"NO\n";
    return 0;
}