KMP 字符串匹配算法

 

#include  " stdafx.h "
#include  < string.h >
#include  < iostream >
#include  < vector >
using namespace std ;

// 模式匹配,kmp算法,复杂度O(m+n)
// 返回匹配位置,-1表示匹配失败,传入匹配串和模式串和长度
// 可更改元素类型,更换匹配函数

#define MAXN  100
#define _match(a,b) ((a) == (b))
typedef  char  elem_t;

int  kmp_match( int  ls,elem_t *  str, int  lp,elem_t *  pat)
... {
    int fail[MAXN]=...{-1}, i=0,j;
    for (j=1;j<lp;)
    ...{
        if(pat[i]!=pat[j] )
        ...{
           fail[j++]=-1 ;
           continue ;
        }
        for(i=0; j<lp && _match(pat[i],pat[j]) ; fail[j++]=i++) ;    
    }

    for (i=j=0;i<ls&&j<lp;i++)
    ...{
        if (_match(str[i],pat[j]))
            j++;
        else if (j)
            j=fail[j-1]+1,i--;
    }
    return j==lp?(i-lp):-1;
}

int  kmp(string  str1, string  str2)
... {
    if(str1.size()<str2.size())
        return -1 ;
    size_t i, j ;
    vector<int> fail ;
    fail.resize(str2.size(),-1) ;

    for(j=1, i=0; j<str2.size(); ++j)
    ...{
        if(str2[j]!=str2[i])
            continue ;
        for(i=0; j<str2.size() && str2[j]==str2[i] ; fail[j++]=i++) ;
        --j ;
    }
    for (i=j=0; i<str2.size() && j<str1.size(); ++j)
    ...{
        if(str1[j]==str2[i])
            ++i ;
        else if(i)
            i=fail[i-1]+1, --j ;
    }
    return i==str2.size() ? j-i :-1 ;
}

int  _tmain( int  argc, _TCHAR *  argv[])
... {
    char * str1="abababababcabababa" ;
    char * str2="ababababcab" ;
    int i=kmp_match(strlen(str1),str1,strlen(str2),str2) ;
    cout<<i<<endl ;
    i=kmp(str1,str2) ;
    cout<<i<<endl ;
    getchar() ;
    return 0;
}