214.shortest-palindrome

Statement

• Difficulty: Hard
• Tag: `字符串` `字符串匹配` `哈希函数` `滚动哈希`

``````输入：s = "aacecaaa"

``````

``````输入：s = "abcd"

``````

• `0 <= s.length <= 5 * 104`
• `s` 仅由小写英文字母组成

• Difficulty: Hard
• Tag: `String` `String Matching` `Hash Function` `Rolling Hash`

You are given a string `s`. You can convert `s` to a palindrome by adding characters in front of it.

Return the shortest palindrome you can find by performing this transformation.

Example 1:

``````Input: s = "aacecaaa"
Output: "aaacecaaa"
``````

Example 2:

``````Input: s = "abcd"
Output: "dcbabcd"
``````

Constraints:

• `0 <= s.length <= 5 * 104`
• `s` consists of lowercase English letters only.

Solution

``````#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>

#define endl "\n"
#define fi first
#define se second
#define all(x) begin(x), end(x)
#define rall rbegin(a), rend(a)
#define bitcnt(x) (__builtin_popcountll(x))
#define complete_unique(a) a.erase(unique(begin(a), end(a)), end(a))
#define mst(x, a) memset(x, a, sizeof(x))
#define MP make_pair

using ll = long long;
using ull = unsigned long long;
using db = double;
using ld = long double;
using VLL = std::vector<ll>;
using VI = std::vector<int>;
using PII = std::pair<int, int>;
using PLL = std::pair<ll, ll>;

using namespace __gnu_pbds;
using namespace std;
template <typename T>
using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
const ll mod = 1e9 + 7;

template <typename T, typename S>
inline bool chmax(T &a, const S &b) {
return a < b ? a = b, 1 : 0;
}

template <typename T, typename S>
inline bool chmin(T &a, const S &b) {
return a > b ? a = b, 1 : 0;
}

#ifdef LOCAL
#include <debug.hpp>
#else
#define dbg(...)
#endif

const int N = 1e5 + 5;

struct Manacher {
int len, l;
char Ma[N << 1];
int Mp[N << 1];

// 0-index
void work(const char *s) {
len = strlen(s);
l = 0;
Ma[l++] = '\$';
Ma[l++] = '#';
for (int i = 0; i < len; ++i) {
Ma[l++] = s[i];
Ma[l++] = '#';
}
Ma[l] = 0;
for (int i = 0; i < l; i++) {
Mp[i] = 0;
}
int mx = 0, id = 0;
for (int i = 0; i < l; ++i) {
Mp[i] = mx > i ? min(Mp[2 * id - i], mx - i) : 1;
while (i - Mp[i] >= 0 && Ma[i + Mp[i]] == Ma[i - Mp[i]]) Mp[i]++;
if (i + Mp[i] > mx) {
mx = i + Mp[i];
id = i;
}
}
}

bool check(int l, int r) {
int il = (l + 1) * 2, ir = (r + 1) * 2;
int mid = (il + ir) / 2;
int len = (r - l + 2) / 2;
return (Mp[mid] / 2) >= len;
}
} man;

class Solution {
public:
string shortestPalindrome(string s) {
man.work(s.c_str());

int m = 0;
for (int i = 0; i < s.length(); i++) {
if (man.check(0, i)) {
m = i + 1;
}
}

auto t = s.substr(m);
reverse(all(t));

return t + s;
}
};

#ifdef LOCAL

int main() {
return 0;
}

#endif
``````