字符串匹配(KMP模板题)

---

（KMP讲解推荐链接:http://www.cnblogs.com/SYCstudio/p/7194315.html）

---

题目:
给定两个由小写字母构成的字符串 L 和 S 。
请你从左到右，找出子串 L 在母串 S 中每次出现的开始位置（匹配位置）。

输入:
第一行：给一个全由小写字母构成的母串 S（0＜S的长度≤1000000）；
第二行：给一个全由小写字母构成的子串 L（0＜L的长度≤S的长度）。

输出:
按升序输出一行一个整数，分别表示子串 L 在母串 S 中每次出现的开始位置。
如果子串 L 在母串 S 中没有出现，则输出“NO”。

EG(1)
in:
yuabcierabcde
abc

out:
3
9

EG(2)
in:
abcdefg
abcdefu

out:
NO

---

代码:

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cctype>
#include<queue>
using namespace std;

const int N=1e6+5;

char a[N],b[N];
int f[N];
bool check=false;

int main()
{
    cin>>a>>b;

    int n=strlen(a);
    int m=strlen(b);
//------------------------------------------------  < f(next)数组的求解 >
    f[0]=-1;

    for(int i=1;i<m;++i)
    {
        int j=f[i-1];

        while(b[j+1]!=b[i]&&j>=0)
            j=f[j];

        if(b[j+1]==b[i]) f[i]=j+1;
        else             f[i]=-1;
    }
//------------------------------------------------  < 比较 > 
    int i=0,j=0;                  

    while(i<n)
    {
        if(a[i]==b[j])
        {
            ++i,++j;

            if(j==m)
                cout<<i-m+1<<endl,j=f[j-1]+1,check=true;
        }
        else
        {
            if(j==0) ++i;
            else     j=f[j-1]+1;
        }
    }
//------------------------------------------------- 
    if(!check)
        cout<<"NO"<<endl;

    return 0;
}