洛谷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

比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 }      

作者：自為風月馬前卒

個人部落格http://attack204.com//

出處：http://zwfymqz.cnblogs.com/

本文版權歸作者和部落格園共有，歡迎轉載，但未經作者同意必須保留此段聲明，且在文章頁面明顯位置給出原文連接配接，否則保留追究法律責任的權利。