1. 程式人生 > >【luogu 3375】【模板】KMP字符串匹配

【luogu 3375】【模板】KMP字符串匹配

col 100% while turn tdi 技術分享 其中 限制 分享

題目描述

如題,給出兩個字符串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<=1000

樣例說明:

技術分享

所以兩個匹配位置為1和3,輸出1、3

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<iostream>
 4 #include<algorithm>
 5 #define maxn 1000006
 6
#define maxm 1003 7 using namespace std; 8 char s1[maxn],s2[maxm];int len1,len2,nxt[maxn],cnt,ans[maxn]; 9 void get_next(char s[]){ 10 nxt[0]=nxt[1]=0; 11 for(int i=1;i<len2;i++){ 12 int j=nxt[i]; 13 while(j&&s[i]!=s[j]) j=nxt[j]; 14 if(s[i]==s[j]) nxt[i+1
]=j+1; 15 else nxt[i+1]=0; 16 } 17 } 18 void kmp(char s1[],char s2[]){ 19 int j=0; 20 for(int i=0;i<len1;i++){ 21 while(j&&s1[i]!=s2[j]) j=nxt[j]; 22 if(s1[i]==s2[j]) j++; 23 if(j==len2) printf("%d\n",i-len2+2); 24 } 25 } 26 int main(){ 27 scanf("%s%s",s1,s2); 28 len1=strlen(s1),len2=strlen(s2); 29 get_next(s2); 30 kmp(s1,s2); 31 for(int i=1;i<=len2;i++)printf("%d ",nxt[i]); 32 return 0; 33 }

【luogu 3375】【模板】KMP字符串匹配