KMP算法板子
阿新 • • 發佈:2017-10-18
mat inline == void using ostream gis nbsp putchar
luogu P3375 【模板】KMP字符串匹配
題目描述
如題,給出兩個字符串s1和s2,其中s2為s1的子串,求出s2在s1中所有出現的位置。
為了減少騙分的情況,接下來還要輸出子串的前綴數組next。
(如果你不知道這是什麽意思也不要問,去百度搜[kmp算法]學習一下就知道了。)
輸入輸出格式
輸入格式:
第一行為一個字符串,即為s1(僅包含大寫字母)
第二行為一個字符串,即為s2(僅包含大寫字母)
輸出格式:
若幹行,每行包含一個整數,表示s2在s1中出現的位置
接下來1行,包括length(s2)個整數,表示前綴數組next[i]的值。
輸入輸出樣例
輸入樣例#1:ABABABC ABA輸出樣例#1:
1 3 0 0 1
說明
時空限制:1000ms,128M
數據規模:
設s1長度為N,s2長度為M
對於30%的數據:N<=15,M<=5
對於70%的數據:N<=10000,M<=100
對於100%的數據:N<=1000000,M<=1000000
樣例說明:
所以兩個匹配位置為1和3,輸出1、3
#include<iostream> #include<cstdio> #include<cstring> #include<string> #define BIG 2000011 using namespace std; char A[BIG],B[BIG]; int lena,lenb; int fail[BIG]; inline void init(){ fail[1]=0; register int j=0; for(register int i=2;i<=lenb;++i){ while(j&&B[j+1]!=B[i]) j=fail[j]; if(B[j+1]==B[i]) ++j; fail[i]=j; } } inline void match(){ register int j=0; for(register int i=1;i<=lena;++i){ while(j&&B[j+1]!=A[i]) j=fail[j]; if(B[j+1]==A[i]) ++j; if(j==lenb) printf("%d\n",i-lenb+1); } } inline void KMP(){ init(); match(); } int main(){ scanf("%s\n%s",A+1,B+1); lena=strlen(A+1); lenb=strlen(B+1); KMP(); for(register int i=1;i<=lenb;++i) printf("%d ",fail[i]); putchar(‘\n‘); return 0; }
KMP算法板子