任何疑问、意见、建议请留言公众号:一航代码
题目描述:
输入两个字符串,分别为模式串和文本串。若模式串是文本串的子字符串,则要求将文本串中的子字符串删除,并返回删除子字符串后的字符串。
输入格式:
首先输入模式串,然后输入文本串。(均由大写或者小写字母构成,二者以空格为间隔隔开)
输出格式:
若模式串是文本串的子字符串,输出删除子字符串后的字符串。
若模式串不是文本串的子字符串,输出-1。
输入样例:
dfg asdfghj
输出样例:
ashj
解决方法:
(1)算法思路:
KMP 算法(Knuth-Morris-Pratt 算法)。
状态机思想参考:动态规划之 KMP 算法详解
(2)代码实现:
#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <math.h>
using namespace std;
vector<vector<int>> KMP(string pat)
{
int M = pat.size();
// dp[状态][字符] = 下个状态
vector<vector<int>> dp(M, vector<int>(256));
// base case
dp[0][pat[0]] = 1;
// 影子状态 X 初始为 0
int X = 0;
// 构建状态转移图(稍改的更紧凑了)
for (int j = 1; j < M; j++)
{
for (int c = 0; c < 256; c++)
{
dp[j][c] = dp[X][c];
dp[j][pat[j]] = j + 1;
// 更新影子状态
X = dp[X][pat[j]];
}
}
return dp;
}
int search(string txt, string pat, vector<vector<int>> dp)
{
int M = pat.size();
int N = txt.size();
// pat 的初始态为 0
int j = 0;
for (int i = 0; i < N; i++)
{
// 计算 pat 的下一个状态
j = dp[j][txt[i]];
// 到达终止态,返回结果
if (j == M)
return i - M + 1;
}
// 没到达终止态,匹配失败
return -1;
}
int main()
{
string pat;//模式串
string txt;//文本串
cin >> pat >> txt;
vector<vector<int>> dp(pat.size(), vector<int>(256));
dp = KMP(pat);//状态机解KMP
int index = search(txt, pat, dp); //匹配成功返回文本串下标
if (index) //否则返回-1
{
string t1 = txt.substr(0, index); //切掉字符串前半部分
string t2 = txt.substr(index + pat.size()); //切掉字符串后半部分
string res = t1 + t2; //组合
cout << res << endl;
}else{
cout<<"-1"<<endl;
}
system("pause");
return 0;
}