1. 程式人生 > >洛谷P3375 【模板】KMP字符串匹配

洛谷P3375 【模板】KMP字符串匹配

ron length efi copy 100% long pac strong htm

題目描述

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

比kmp好理解的多

把b串做成一個hash值

在a串中不斷的匹配就好

時間復雜度:O(n)O(n)

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4
#define ull unsigned long long 5 using namespace std; 6 const int MAXN=2e7+10; 7 const int mod=19260817; 8 const int seed=233; 9 ull po[MAXN]; 10 ull hash(ull *a,int l,int r) 11 { 12 if(l-1<0) return a[r]; 13 return a[r]-po[r-l+1]*a[l-1]; 14 } 15 char s1[MAXN],s2[MAXN];
16 ull h1[MAXN],h2; 17 int l1,l2; 18 int nxt[MAXN]; 19 int main() 20 { 21 po[0]=1; 22 for(int i=1;i<=10001;i++) po[i]=po[i-1]*seed; 23 scanf("%s",s1);scanf("%s",s2); 24 l1=strlen(s1);l2=strlen(s2); 25 for(int i=0;i<l1;i++) h1[i]=h1[i-1]*seed+s1[i]; 26 for(int i=0;i<l2;i++) h2=h2*seed+s2[i]; 27 for(int i=0;i<l1;i++) 28 if(hash(h1,i,i+l2-1)==h2) 29 printf("%d\n",i+1); 30 31 int j=0; 32 for(int i=1;i<l2;i++) 33 { 34 while(j>0&&s2[i]!=s2[j]) j=nxt[j-1]; 35 if(s2[i]==s2[j]) j++; 36 nxt[i]=j; 37 }//題目中說要輸出nxt數組 38 for(int i=0;i<l2;i++) printf("%d ",nxt[i]); 39 return 0; 40 }

洛谷P3375 【模板】KMP字符串匹配