SCU 4495 单词替换 KMP

最新推荐文章于 2020-11-06 14:06:48 发布

swust_fangfangfang

最新推荐文章于 2020-11-06 14:06:48 发布

阅读量423

点赞数

分类专栏： KMP 文章标签： KMP

本文链接：https://blog.csdn.net/ftqooo/article/details/51900777

版权

KMP 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

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[N];  ///原字符串
int nxt[N];
bool vis[N];///查找到了进行标记的数组
char A[N];  ///查找的字符串
char B[N];  ///替换的字符串
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;
}