KMP板子

Posted 2023-05-02 x383494

P3426

#include <cstdio>
#include <cstring>
#include <vector>
#define sd std::
namespace m // 
constexpr int LEN = 1e6;
sd vector<int> prepare(char* a)
	int len = strlen(a);
	sd vector<int> ret(len);
	for(int i=1; a[i]; ++i)
		int j = ret[i-1];
		while(j && a[i] != a[j]) j = ret[j-1];
		if(a[i] == a[j]) ++j;
		ret[i] = j;
	
	return ret;

char s1[LEN+1], s2[LEN+1];
void work()
	scanf("%s%s", s1, s2);
	auto pi = prepare(s2);
	int len2 = strlen(s2);
	int j = 0;
	for(int i=0; s1[i]; ++i)
		while(j && s1[i] != s2[j]) j = pi[j-1];
		if(s1[i] == s2[j]) ++j;
		if(j == len2) printf("%d\\n", i-len2+2), j = pi[j-1];
	
	for(int i:pi)
		printf("%d ", i);
	

 int main()m::work(); return 0;

KMP板子题

 

 

 

https://www.luogu.org/problem/P3375

 

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;
int nxt[maxn];
int f[maxn];
char a[maxn];
char b[maxn];
int lena,lenb;
int ans[maxn],cnt;
void getnxt() // a是子串b是主串
{
    nxt[1]=0;
    for(int i=2,j=0; i<=lena; i++)
    {
        while(j>0&&a[i]!=a[j+1]) j=nxt[j];
        if(a[i]==a[j+1]) j++;
        nxt[i]=j;
    }
}
void kmp()
{
    int i,j;
    for(i=1,j=0; i<=lenb; i++)
    {
        while(j>0&&(j==lena||b[i]!=a[j+1])) j=nxt[j];
        if(b[i]==a[j+1]) j++;
        if(j==lena)
            ans[++cnt]=i-lena+1;
    }
}
int main()
{
    scanf("%s%s",b+1,a+1);
    lena=strlen(a+1);
    lenb=strlen(b+1);
    getnxt();
    kmp();
    for(int i=1; i<=cnt; i++)
        printf("%d
",ans[i]);
    for(int i=1; i<=lena; i++)
        printf("%d ",nxt[i]);
}