[2125]串结构练习——字符串匹配 sdutOJ



串结构练习——字符串匹配

Time Limit: 1000ms   Memory limit: 65536K  有疑问？点这里^_^

题目描述

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

 

输入

 输入包含多组数据，每组测试数据包含两行，第一行代表string1，第二行代表string2，string1和string2中保证不出现空格。

 

输出

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

 

示例输入

abc
a
123456
45
abc
ddd

示例输出

YES
YES
NO

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
int next[1000010];
char a[1000010],b[1000010];
void get_next()//next数组
{
    int i=0;
    int j=-1;
    next[0]=-1;
    int len=strlen(b);
    while(i<len)
    {
        if(j==-1||b[i]==b[j])
        {
            i++;
            j++;
            next[i]=j;
        }
        else
            j=next[j];
    }
}
void KMP()//KMP算法的运用
{
    int i=0,j=0;
    int len1=strlen(a);
    int len2=strlen(b);
    while(i<len1&&j<len2)
    {
        if(j==-1||a[i]==b[j])
        {
            i++;
            j++;
        }
        else
            j=next[j];
    }
    if(j>=len2)
        printf("YES\n");
    else
        printf("NO\n");

}
int main()
{
    while(~scanf("%s %s",a,b))
    {
        get_next();
        KMP();
    }

    return 0;
}