扩展kmp模板

扩展kmp详解：扩展 KMP 算法

//辅助数组next[i]表示T[i,m-1]和T的最长公共前缀长度
//设置两个变量，a 和 p。p 代表以 a 为起始位置的字符匹配成功的最右边界，也就是 “p = 最后一个匹配成功位置 + 1”。
模板：P5410 【模板】扩展 KMP

#include <stdio.h>
#include <math.h>
#include <string.h>
#include <algorithm>
#include <stack>
using namespace std;
const int manx=1e5+10;
char s[manx],pp[manx];
int extend[manx],net[manx],ls,lp;
void getnext()
{
    int a=0,p=0;
    net[0]=lp;
    for(int i=1; i<lp; i++)
    {
        if(i>=p||i+net[i-a]>=p)
        {
            if(i>=p)
                p=i;
            while(p<lp&&pp[p]==pp[p-i])
                p++;
            net[i]=p-i;
            a=i;
        }
        else
            net[i]=net[i-a];
    }
}
void ex_kmp()
{
    getnext();
    int a=0,p=0;
    for(int i=0; i<ls; i++)
    {
        if(i>=p||i+net[i-a]>=p)
        {
            if(i>=p)
                p=i;
            while(p<ls&&p-i<lp&&s[p]==pp[p-i])
                p++;
            extend[i]=p-i;
            a=i;
        }
        else
            extend[i]=net[i-a];
    }
}
int main()
{
    while(scanf("%s%s",s,pp)!=EOF)
    {
        ls=strlen(s);
        lp=strlen(pp);
        ex_kmp();
        for(int i=0; i<ls-1; i++)
            printf("%d ",extend[i]);
        printf("%d\n",extend[ls-1]);
    }
    return 0;
}

例题：HDU 4333 Revolving Digits
因为题目要求的是不同数的个数，所以要去重，同一个数算过的次数也就是最小循环节出现的次数。

//统计最小循环节出现的次数
for(int i = 0; i < lp;i++)
    if(extend[i] == lp)
        cnt++;