SCU 4495 单词替换 KMP

Time Limit: 500 MS    Memory Limit: 64000 K

Description

给出一个仅包含小写字母的字符串s，和单词A，B。把s中所有的出现过的A替换为B。

Input

第一行一个数T(1<=T<=10)，表示数据组数
每组数据三行，第一行为s，第二行为A，第三行为B。所有字符串仅包含小写字母
且长度小于5,000,000。

Output

每组数据输出一行，替换后的字符串。

Sample Input

3
aaa
a
b
aaa
aa
b
ababa
aba
cd

Sample Output

bbb
ba
cdba

解法：模版KMP，我用的大白书模版

这道题就用kmp对找到的单词第一个字母进行标记，一个字符一个字符的输出，到了标记了的地方就输出B字符串，左边变成i+strlen(A)就行了。

我理解的KMP：

1、首先是对寻找的字串的操作，用一个nxt记录，nxt[i+1]里面记录是的以当前i字符为终点，它之前能组成的与第一个字符开始的最长相同前缀长度。即长度相同的前缀和后缀

例：a b c a b y

 nxt  0 0 0 0 1 2 0

是一个斜对应的数组。

2、然后就是匹配的过程，用S代表原字符串，A代表需要查找的字符串，在顺序匹配过程中检测到S某个字符与A中第一个字符匹配上了，进行暴力匹配，发现有一个字符不匹配。这时候就需要使用nxt数组了，nxt记录的数字就可以跳到S后一个字符与A相同前缀的后一个字符进行匹配。这样就节约了时间。

感觉不好理解

例：  S:abcabcaby      A:abcaby

匹配的过程：

估计没人看得懂，还是看B站的吧（我在B站学算法）：点击打开链接

CODE

#include <bits/stdc++.h>
using namespace std;
const int N = 5000000+10;
char s
; ///原字符串
int nxt
;
bool vis
;///查找到了进行标记的数组
char A
; ///查找的字符串
char B
; ///替换的字符串
void getFail(){ ///预处理nxt数组
int len = strlen(A);
nxt[0] = nxt[1] = 0;
for(int i = 1;i < len;i++){
int j = nxt[i];
while(j && A[i] != A[j]) j = nxt[j];
nxt[i+1] = (A[i] == A[j] ? j+1:0);
}
}

void Find(){ ///字符串匹配
int lens = strlen(s),lenA = strlen(A);
getFail();
int j = 0;
for(int i = 0;i < lens;i++){ ///大白书模版
while(j && A[j] != s[i])
j = nxt[j];
if(A[j] == s[i]) j++;
if(j == lenA) vis[i-lenA+1] = true;
}
}

int main(void)
{
int T;
scanf("%d",&T);
while(T--){
memset(vis,false,sizeof vis);
memset(nxt,0,sizeof nxt);
scanf("%s%s%s",s,A,B);
Find();
int lens = strlen(s),lenA = strlen(A);
for(int i = 0;i < lens;){
if(!vis[i]) printf("%c",s[i++]);
else{
printf("%s",B);
i = i+lenA;
}
}
puts("");
}
return 0;
}