字符串_马拉车算法
马拉车算法与kmp算法的优化方法基本一样,就是把已经求出来的回文片段用来算接下来要求的回文中心(来回颠倒),在此附上一道马拉车算法题目的代码。
点击查看代码
#include <bits/stdc++.h>
using namespace std;
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int t;
cin >> t;
while (t--) {
string s;
int n;
cin >> n;
cin >> s;
vector<int> d1(n), d2(n);
int L = 0, R = -1;
for (int i = 0; i < n; i++) {
int k;
if (i > R) {
k = 1;
} else {
int j = L + R - i;
k = min(d1[j], R - i + 1);
}
while (i - k >= 0 && i + k < n && s[i - k] == s[i + k]) {
k++;
}
d1[i] = k;
if (i + k - 1 > R) {
L = i - k + 1;
R = i + k - 1;
}
}
L = 0, R = -1;
for (int i = 0; i < n; i++) {
int k;
if (i > R) {
k = 0;
} else {
int j = L + R - i + 1;
k = min(d2[j], R - i + 1);
}
while (i - k - 1 >= 0 && i + k < n && s[i - k - 1] == s[i + k]) {
k++;
}
d2[i] = k;
if (i + k - 1 > R) {
L = i - k;
R = i + k - 1;
}
}
vector<int> le(n), ri(n);
for (int i = 0; i < n; i++) {
if (i + d1[i] - 1 < n) {
ri[i + d1[i] - 1] = max(d1[i] * 2 - 1, ri[i + d1[i] - 1]);
}
if (i + d2[i] - 1 < n && i + d2[i] - 1 >= 0) {
ri[i + d2[i] - 1] = max(d2[i] * 2, ri[i + d2[i] - 1]);
}
if (i - d1[i] + 1 >= 0) {
le[i - d1[i] + 1] = max(d1[i] * 2 - 1, le[i - d1[i] + 1]);
}
if (i - d2[i] >= 0) {
le[i - d2[i]] = max(d2[i] * 2, le[i - d2[i]]);
}
}
for (int i = 0; i + 1 < n; i++) {
le[i + 1] = max(le[i + 1], le[i] - 2);
}
for (int i = n - 1; i > 0; i--) {
ri[i - 1] = max(ri[i - 1], ri[i] - 2);
}
int ans = 0;
for (int i = 0; i < n - 1; i++) {
ans = max(ri[i] + le[i + 1], ans);
}
cout << ans << '\n';
}
return 0;
}
浙公网安备 33010602011771号