KMP与拓展KMP

！友情提示：next数组建议写成Next，否则会CE

1.kmp模板

#include<iostream>
#include<cstring>
using namespace std;
const int maxn=10005;
char p[maxn],t[maxn];
int next[maxn];
void get_next(){
    int len=strlen(t);
    int i=0,j=-1;
    next[0]=-1;
    while(i<len){
        if(j==-1||t[i]==t[j]){
            i++;
            j++;
            next[i]=j;
        }
        else j=next[j];
    }
}

bool kmp(){
    int i=0,j=0;
    int len1=strlen(p);
    int len2=strlen(t);
    while(i<len1&&j<len2){
        if(j==-1||p[i]==t[j]){
            i++;
            j++;
        }
        else j=next[j];
    }
    if(j==len2)return true;
    else return false;
}

int main(){
    cin>>p;
    cin>>t;
    get_next();
    if(kmp())cout<<"YES"<<endl;
    else cout<<"NO"<<endl;
    return 0;
}

2.拓展kmp（ex数组存储的是以s1各元素起始的字串与s2前缀的最大匹配长度）

const int maxn=505;   //字符串长度最大值
char a[101][maxn];
int len[maxn];
int Next[maxn],ex[maxn]; //ex数组即为extend数组

//预处理计算next数组
void GETNEXT(char *str)
{
    int i=0,j,po,len=strlen(str);
    Next[0]=len;//初始化next[0]
    while(str[i]==str[i+1]&&i+1<len)//计算next[1]
    i++;
    Next[1]=i;
    po=1;//初始化po的位置
    for(i=2;i<len;i++)
    {
        if(Next[i-po]+i<Next[po]+po)//第一种情况，可以直接得到next[i]的值
        Next[i]=Next[i-po];
        else//第二种情况，要继续匹配才能得到next[i]的值
        {
            j=Next[po]+po-i;
            if(j<0)j=0;//如果i>po+Next[po],则要从头开始匹配
            while(i+j<len&&str[j]==str[j+i])//计算next[i]
            j++;
            Next[i]=j;
            po=i;//更新po的位置
        }
    }
}
//计算extend数组
void EXKMP(char *s1,char *s2)
{
    int i=0,j,po,len=strlen(s1),l2=strlen(s2);
    GETNEXT(s2);//计算子串的next数组
    while(s1[i]==s2[i]&&i<l2&&i<len)//计算ex[0]
    i++;
    ex[0]=i;
    po=0;//初始化po的位置
    for(i=1;i<len;i++)
    {
        if(Next[i-po]+i<ex[po]+po)//第一种情况，直接可以得到ex[i]的值
        ex[i]=Next[i-po];
        else//第二种情况，要继续匹配才能得到ex[i]的值
        {
            j=ex[po]+po-i;
            if(j<0)j=0;//如果i>ex[po]+po则要从头开始匹配
            while(i+j<len&&j<l2&&s1[j+i]==s2[j])//计算ex[i]
            j++;
            ex[i]=j;
            po=i;//更新po的位置
        }
    }
}

拓展KMP题目（HDU - 1238）

题意：给定x串，在这x个串中找出一个字串，使得这个字串或这个字串的倒置在所有的串中都出现过，求这个字串的最大长度。

代码：
 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <vector>
#include <algorithm>
#define pb(x) push_back(x)
#define me(x,y) memset(x,y,sizeof(x))
#define sd(x) scanf("%d",&x)
#define ss(x) scanf("%s",x)
#define sf(x) scanf("%f",&x)
#define slf(x) scanf("%lf",&x)
#define slld(x) scanf("%lld",&x)
#define pd(x) printf("%d\n",x)
#define pc(x) printf("%c",x)
#define pn() printf("\n")
#define plld(x) printf("%lld\n",x)
#define ps(x) printf("%s\n",x)
#define max(x,y) (x>=y?x:y)
#define min(x,y) (x<y?x:y)
#define sum(x,y) (x+y)
#define INF 0x3f3f3f3f

using namespace std;
typedef long long ll;
typedef unsigned long long ull;

const int maxn=505;   //字符串长度最大值
char a[101][maxn];
int len[maxn];
int Next[maxn],ex[maxn]; //ex数组即为extend数组

//预处理计算next数组
void GETNEXT(char *str)
{
    int i=0,j,po,len=strlen(str);
    Next[0]=len;//初始化next[0]
    while(str[i]==str[i+1]&&i+1<len)//计算next[1]
    i++;
    Next[1]=i;
    po=1;//初始化po的位置
    for(i=2;i<len;i++)
    {
        if(Next[i-po]+i<Next[po]+po)//第一种情况，可以直接得到next[i]的值
        Next[i]=Next[i-po];
        else//第二种情况，要继续匹配才能得到next[i]的值
        {
            j=Next[po]+po-i;
            if(j<0)j=0;//如果i>po+Next[po],则要从头开始匹配
            while(i+j<len&&str[j]==str[j+i])//计算next[i]
            j++;
            Next[i]=j;
            po=i;//更新po的位置
        }
    }
}
//计算extend数组
void EXKMP(char *s1,char *s2)
{
    int i=0,j,po,len=strlen(s1),l2=strlen(s2);
    GETNEXT(s2);//计算子串的next数组
    while(s1[i]==s2[i]&&i<l2&&i<len)//计算ex[0]
    i++;
    ex[0]=i;
    po=0;//初始化po的位置
    for(i=1;i<len;i++)
    {
        if(Next[i-po]+i<ex[po]+po)//第一种情况，直接可以得到ex[i]的值
        ex[i]=Next[i-po];
        else//第二种情况，要继续匹配才能得到ex[i]的值
        {
            j=ex[po]+po-i;
            if(j<0)j=0;//如果i>ex[po]+po则要从头开始匹配
            while(i+j<len&&j<l2&&s1[j+i]==s2[j])//计算ex[i]
            j++;
            ex[i]=j;
            po=i;//更新po的位置
        }
    }
}

int main() {
    int t;
    sd(t);
    while(t--) {
        int n;
        int maxx=0;
        sd(n);
        ss(a[0]);
        len[0]=strlen(a[0]);

        for(int i=0;i<len[0]-1;i++) a[0][len[0]+i]=a[0][len[0]-2-i];
        len[0]=len[0]+len[0]-1;
        a[0][len[0]]='\0';

        for(int i=1;i<n;i++) {
            ss(a[i]);
            len[i]=strlen(a[i]);
        }

        for(int i=0;i<len[0];i++) {
            int ans=INF;
            for(int j=1;j<n;j++) {
                int cnt=0;
                EXKMP(a[j],a[0]+i);
                for(int k=0;k<len[j];k++) cnt=max(cnt,ex[k]);

                ans=min(ans,cnt);
            }
        maxx=max(maxx,ans);
        }
        pd(maxx);
    }
	return 0;
}