字符串匹配(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;
}