Manacher马拉车算法--求最大回文子串

转载：
作者：windliang
链接：链接
来源：力扣（LeetCode）
看了这位大佬的题解明白了马拉车

---

求原字符串下标

用 P 的下标 i 减去 P [ i ]，再除以 2，就是原字符串的开头下标了。

例如我们找到 P[ i ] 的最大值为 5，也就是回文串的最大长度是 5，对应的下标是 6，所以原字符串的开头下标是（6 - 5 ）/ 2 = 0。所以我们只需要返回原字符串的第 0 到 第（5 - 1）位就可以了。

求每个 P [ i ]

接下来是算法的关键了，它充分利用了回文串的对称性。

我们用 C 表示回文串的中心，用 R 表示回文串的右边半径。所以 R = C + P[ i ]。C 和 R 所对应的回文串是当前循环中 R 最靠右的回文串。

让我们考虑求 P [ i ] 的时候，如下图。

用 i_mirror 表示当前需要求的第 i 个字符关于 C 对应的下标。

我们现在要求 P [ i ]，如果是用中心扩展法，那就向两边扩展比对就行了。但是我们其实可以利用回文串 C 的对称性。i 关于 C 的对称点是 i_mirror，P [ i_mirror ] = 3，所以 P [ i ] 也等于 3。

但是有三种情况将会造成直接赋值为 P [ i_mirror ] 是不正确的，下边一一讨论。

1. 超出了 R

当我们要求 P [ i ] 的时候，P [ mirror ] = 7，而此时 P [ i ] 并不等于 7，为什么呢，因为我们从 i 开始往后数 7 个，等于 22，已经超过了最右的 R，此时不能利用对称性了，但我们一定可以扩展到 R 的，所以 P [ i ] 至少等于 R - i = 20 - 15 = 5，会不会更大呢，我们只需要比较 T [ R+1 ] 和 T [ R+1 ]关于 i 的对称点就行了，就像中心扩展法一样一个个扩展。

2. P [ i_mirror ] 遇到了原字符串的左边界

此时P [ i_mirror ] = 1，但是 P [ i ] 赋值成 1 是不正确的，出现这种情况的原因是 P [ i_mirror ] 在扩展的时候首先是 “#” == “#”，之后遇到了 “^” 和另一个字符比较，也就是到了边界，才终止循环的。而 P [ i ] 并没有遇到边界，所以我们可以继续通过中心扩展法一步一步向两边扩展就行了。

3. i 等于了 R

此时我们先把 P [ i ] 赋值为 0，然后通过中心扩展法一步一步扩展就行了。

C 和 R 的更新

就这样一步一步的求出每个 P [ i ]，当求出的 P [ i ] 的右边界大于当前的 R 时，我们就需要更新 C 和 R 为当前的回文串了。因为我们必须保证 i 在 R 里面，所以一旦有更右边的 R 就要更新 R。

此时的 P [ i ] 求出来将会是 3，P [ i ] 对应的右边界将是 10 + 3 = 13，所以大于当前的 R，我们需要把 C 更新成 i 的值，也就是 10，R 更新成 13。继续下边的循环。

时间复杂度：for 循环里边套了一层 while 循环，难道不是 O(n²)？不！其实是 O(n)。不严谨的想一下，因为 while 循环访问 R 右边的数字用来扩展，也就是那些还未求出的节点，然后不断扩展，而期间访问的节点下次就不会再进入 while 了，可以利用对称得到自己的解，所以每个节点访问都是常数次，所以是 O ( n )。

空间复杂度：O(n)。

原作者代码

public String preProcess(String s) {
    int n = s.length();
    if (n == 0) {
        return "^$";
    }
    String ret = "^";
    for (int i = 0; i < n; i++)
        ret += "#" + s.charAt(i);
    ret += "#$";
    return ret;
}

// 马拉车算法
public String longestPalindrome2(String s) {
    String T = preProcess(s);
    int n = T.length();
    int[] P = new int[n];
    int C = 0, R = 0;
    for (int i = 1; i < n - 1; i++) {
        int i_mirror = 2 * C - i;
        if (R > i) {
            P[i] = Math.min(R - i, P[i_mirror]);// 防止超出 R
        } else {
            P[i] = 0;// 等于 R 的情况
        }

        // 碰到之前讲的三种情况时候，需要利用中心扩展法
        while (T.charAt(i + 1 + P[i]) == T.charAt(i - 1 - P[i])) {
            P[i]++;
        }

        // 判断是否需要更新 R
        if (i + P[i] > R) {
            C = i;
            R = i + P[i];
        }

    }

    // 找出 P 的最大值
    int maxLen = 0;
    int centerIndex = 0;
    for (int i = 1; i < n - 1; i++) {
        if (P[i] > maxLen) {
            maxLen = P[i];
            centerIndex = i;
        }
    }
    int start = (centerIndex - maxLen) / 2; //最开始讲的求原字符串下标
    return s.substring(start, start + maxLen);
}

我的代码

class Solution {
public:
    string get_new_str(string s) {
        string new_s = "#";
        for (int i = 0; i < s.size(); i++) {
            new_s += s[i];
            new_s += "#";
        }
        return new_s;
    }
    string longestPalindrome(string s) {
        string  newstr = get_new_str(s);
        int r[3000] = {0}, ind;
        int c = 0, p = 0, maxlen = 0;
        for (int i = 0; i < newstr.size(); i++) {
            if (i <= p) r[i] = min(r[2 * c - i], p - i);
            while (i > r[i] && newstr[i + r[i] + 1] == newstr[i - r[i] - 1]) r[i]++; //得保证i - r[i] - 1 >= 0
            if (i + r[i] > p) {
                c = i;
                p = i + r[i];
            }
            if (r[i] > maxlen) {
                maxlen = max(maxlen, r[i]);
                ind = i;
            }
        }
        //cout << maxlen << endl;
        string ret;
        for (int i = ind - maxlen + 1; i <= ind + maxlen - 1; i += 2) {
            ret += newstr[i];
        }
        return ret;
    }
};

例题

HDOJ 3068 最长回文

#include <iostream>
#include <string>
#include <cstdio>
#define MAX_N 200000

using namespace std;
char str[MAX_N];
string newstr;

string getNewStr(char s[]) {
    string newstr = "#";
    for (int i = 0; s[i]; i++) {
        newstr += s[i];
        newstr += "#";
    }
    return newstr;
}

int manacher(char str[]) {
  	newstr = getNewStr(str);
  	//cout << newstr << " " << newstr.size() << endl;
    int r[MAX_N + 5] = {0};
    int c = 0, p = 0, maxlen = 0;
    for (int i = 0; i < newstr.size(); i++) {
        if (i <= p) r[i] = min(r[2 * c - i], p - i);
       	while (i > r[i] && newstr[i + r[i] + 1] == newstr[i - r[i] - 1]) r[i]++; //得保证i - r[i] - 1 >= 0
   	   	if (i + r[i] > p) {
   	        c = i;
  		  	p = i + r[i];
     	}
     	maxlen = max(maxlen, r[i]);
    }
    return maxlen;
}
int main() {
    while (scanf("%s", str) != EOF){
        //cout << str << endl;
        cout << manacher(str) << endl;
    }
    return 0;
}