KMP

全部代码

#include <stdio.h>
#include <assert.h>
#include <stdlib.h>
#include <string.h>

int *GetNext(const char *match)
{
    int *pNext = NULL;
    int i;
    int j;

    assert(match!=NULL);

    pNext = (int *)malloc(strlen(match)*sizeof(int));
    if(NULL == pNext)
    {
        printf("pNext空间分配失败！
");
        exit(-1);
    }
    i = 1;
    j = i-1;
    //起始位置
    pNext[0] = 0;

    while(i < strlen(match))
    {
        //当前字符与前面的next值所对应位置的值相等
        if(match[i] == match[pNext[j]])
        {
            pNext[i] = pNext[j]+1;
            ++i;
            j = i-1;
        }
        //前一个next值为0
        else if(0 == pNext[j])
        {
            pNext[i] = 0;
            ++i;
            j = i-1;
        }
        //不满足以上情况  向前跳转  再进行比较
        else
        {
            j = pNext[j] - 1;
        }
    }

    return pNext;
}

int KMP(const char *src, const char *match)
{
    int i;
    int j;
    int *pNext = NULL;

    assert(src!=NULL && match!=NULL);

    pNext = GetNext(match);
    i = 0;
    j = 0;

    while(i<strlen(src) && j<strlen(match))
    {
        //相等  二者均向后移动
        if(src[i] == match[j])
        {
            ++i;
            ++j;
        }
        //不相等
        else
        {
            //匹配串已经回到起始位置 且仍不相等 主串向后移动
            if(0 == j)
            {
                ++i;
            }
            //不满足上述情况  匹配串向前跳转
            else
            {
                j = pNext[j-1];
            }
        }
    }

    if(0 == match[j])
    {
        return i - strlen(match);
    }

    return -1;
}

int main(void)
{
    char *src = "ababcabcdef";
    char *match = "abc";
    printf("%d
", KMP(src, match));

    return 0;
}