题目传送门

解题思路:

首先说KMP的作用:对于两个字符串A,B(A.size() > B.size()),求B是否是A的一个字串或B在A里的位置或A里有几个B,说白了就是字符串匹配.

下面创设一个问题背景 : 有两个字符串A,B,求B在A中的位置.

A : abcdcbd

B: cdcb

对于上述问题,我们最先想到也就是最暴力的办法,就是把B整体从1~7每次一格的往右挪,每往挪一格,便把B从头到尾跟A匹配一遍,直到匹配不成功或全部匹配成功,如果匹配不成功,就再整体往右挪一格,这样的时间复杂度最坏可以达到O(nm).

如果字符串再长一点,长度达到1e6,显然暴力是会TLE的,那我们就要考虑优化,上述做法唯一能优化的地方就是往右挪的方式,因为一格一格的挪实在是太笨了,所以我们就要想怎样才能挪的快而且正确呢?

很容易想到的是1.直接移到B的后一位 2.移到失配的位置 ,而这两种做法都是可以被推翻的,见图:

那我们就只能甘于暴力吗?不,所以我们要学习KMP,进入正题:

对于学习KMP,要明白几个重要概念:

对于一个字符串A': abcddc

1.前缀:a,ab,abc,abcd,abcdd.

2.后缀:c,dc,ddc,cddc,bcddc.(前缀和后缀都不包括其本身)

然后我们来看一下KMP的伪代码:

1 while(i < a.length() && j < b.length()) {
2     if(当前位置匹配) i++,j++;
3     else j = kmp[j];  //跳到一个特定的位置 
4 }
5 if(j == b.length()) 成功
6 else 失败 

看完后发现其实KMP的核心,就是求kmp[j](大多数人称为next[j]),下面就来说一下kmp[j]怎么求.

kmp[i]表示的是长度为i(1~i)的字符串的最长公共前后缀的长度.(以下的j皆为不包括本次字符求得的最长公共前后缀长度,其实也就是枚举前缀的长度)

例: P: a b c d a b c 

kmp[]:0 0 0 0 1 2 3

求当前kmp[i],可以分两种情况:

1.当P[i] == P[j+1]时(见下图),kmp[i] = j+1,因为j=2说明P[1..2]和P[3..4]是完全相等的,而P[5]==P[3],那么P[1..3]和P[3..5]是完全相等的,那kmp[5] = 3;

2.当P[i] != P[j+1]时(见下图),j = kmp[j]往回跳,因为j == 3,说明P[1..3]和P[5..7]完全相等,而后缀的最后加上一个P[i],则意味着我们相当于在P[4]插入一个P[i],要在P[1..3]找一个位置j,使P[i] == P[j+1],转化为第一种情况,求kmp[4],即为kmp[i].那为什么j=kmp[j]呢,举个例子:aba插入一个b,只有两个a相等,后面才有可能成为公共前后缀.还有就是j要跳到什么程度呢?答案就是,当找到P[i]==P[j+1]时,或j跳到0了,说明不可能形成公共前后缀了,就停止.

下面上求kmp[]的代码:

1 string l;
2 j = 0;
3 kmp[1] = 0;
4 for(int i = 2;i <= n; i++) {
5     while(j != 0 && l[i] != l[j+1]) j = kmp[j];
6     if(l[i] == l[j+1]) j++;
7     kmp[i] = j;
8 }

然后,我们再回归问题,如何在A串中匹配B呢?Talk is cheap,show you the code.

 1 string a,b;
 2 j = 0;
 3 kmp[1] = 0;
 4 for(int i = 2;i <= n; i++) {//n为B的长度 
 5     while(j != 0 && b[i] != b[j+1]) j = kmp[j];
 6     if(b[i] == b[j+1]) j++;
 7     kmp[i] = j;
 8 }
 9 j = 0;
10 for(int i = 1;i <= m; i++) {//m为A的长度 
11     while(j != 9 && a[i] != b[j+1]) j = kmp[j];//匹配失败就往回跳 
12     if(a[i] == b[j+1]) j++;//当前匹配成功 
13     if(j == n) {//整个B匹配成功 
14         printf("%d",i - j + 1);
15         return 0;
16     }
17 }

最后说一点,就是对于A和B匹配的时候,并不是i在向后动,而是j在向前动.

最后的吐槽时间:这算法好难理解(自认为),想出这算法的三个人也太nb了,写这篇博客好累,虽然不一定有人看,但心里还是挺高兴的.

AC代码:

//洛谷题的AC代码

 1 #include<iostream>
 2 #include<cstdio>
 3 
 4 using namespace std;
 5 
 6 string l1,l,pp,p2;
 7 int lena,lenb,next[1000001],j;
 8 
 9 int main() {
10     l1 = l = " ";
11     cin >> pp >> p2;
12     l += pp;l1 += p2;
13     lena = l.length();
14     lenb = l1.length();
15     next[1] = 0;
16     for(int i = 2;i < lenb; i++) {
17         while(j != 0 && l1[i] != l1[j+1]) j = next[j];
18         if(l1[j+1] == l1[i]) j++;
19         next[i] = j;
20     }
21     j = 0;
22     for(int i = 1;i < lena; i++) {
23         while(j != 0 && l[i] != l1[j+1]) j = next[j];
24         if(l[i] == l1[j+1]) j++;
25         if(j == lenb - 1) {
26             printf("%d\n",i - lenb + 2);
27             j = next[j];
28         }
29     }
30     for(int i = 1;i < lenb; i++)
31         printf("%d ",next[i]);
32     return 0;
33 }

 

标签：洛谷,..,int,kmp,next,++,&&,l1,P3375
来源： https://www.cnblogs.com/lipeiyi520/p/12359489.html