KMP及EXKMP模板

kmp

#include<iostream>
#include<algorithm>
#include<string.h>
#include<map>
#include<queue>
#include<cmath>
#include<cstdio>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int maxn=2e5+5;
int Next[maxn];
void getnext(char *ptr)
{
 
    int i,n,k;
    n=strlen(ptr);
    k=-1;
    Next[0]=-1;
    i=0;
    while(i<n)
    {
        if(k==-1 || ptr[i]==ptr[k])
        {
            i++;
            k++;
            Next[i]=k;
        }
        else k=Next[k];
    }
}
int kmp(char *a,char *b)//查找子串出现个数
{
    int i=0,j=0,ans=0;
    int len1=strlen(a);
    int len2=strlen(b);
    getnext(b);
    while(i<len1)
    {
      if(j==-1||a[i]==b[j])
        ++i,++j;
      else
        j=Next[j];
      if(j==len2)//匹配到一个了就当没找到然后跳到next[J] 
		{
			j=Next[j];
			ans++;
		}
    }
    return ans;
}
/*bool kmp(char *a,char *b)//单纯查找是不是子串
{
    int i=0,j=0;
    int len1=strlen(a);
    int len2=strlen(b);
    getnext(b);
    while(i<len1&&j<len2)
    {
      if(j==-1||a[i]==b[j])
        ++i,++j;
      else
        j=Next[j];
    }
    if(j==len2)
    	return  1;
    else
        return 0;
}*/
int main()
{
	
}

EXKMP

#include<iostream>
#include<algorithm>
#include<string.h>
#include<map>
#include<queue>
#include<cmath>
#include<cstdio>
#include<stack>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int maxn=5e4+5;
const int MOD=1e9+7;
int nex[maxn];
int extend[maxn];
void pre_exkmp(char *b)
{
    int len=strlen(b);
    nex[0]=len;
    int j=0;
    while(j+1<len&&b[j]==b[j+1]) j++;
    nex[1]=j;
    int k=1;
    for(int i=2;i<len;i++)
    {
        int p=nex[k]+k-1;
        int L=nex[i-k];
        if(i+L<p+1) nex[i]=L;
        else
        {
            j=max(0,p-i+1);
            while(i+j<len&&b[i+j]==b[j]) j++;
            nex[i]=j;
            k=i;
        }
    }
}
void exkmp(char *a,char *b)
{
    pre_exkmp(b);
    int j=0;
    int lena=strlen(a);
    int lenb=strlen(b);
    while(j<lena&&j<lenb&&a[j]==b[j]) j++;
    extend[0]=j;
    int k=0;
    for(int i=1;i<lena;i++)
    {
        int p=extend[k]+k-1;
        int L=nex[i-k];
        if(i+L<p+1) extend[i]=L;
        else
        {
            j=max(0,p-i+1);
            while(i+j<lena&&j<lenb&&a[i+j]==b[j]) j++;
            extend[i]=j;
            k=i;
        }
    }
}
char a[maxn];//主串
char b[maxn];