[LeetCode] Shortest Palindrome

Given a string S, you are allowed to convert it to a palindrome by adding characters in front of it. Find and return the shortest palindrome you can find by performing this transformation.

For example:

Given "aacecaaa", return "aaacecaaa".

Given "abcd", return "dcbabcd".

Credits:
Special thanks to @ifanchu for adding this problem and creating all test cases. Thanks to @Freezen for additional test cases.

https://leetcode.com/problems/shortest-palindrome/

其實問題可以轉化成求從s[0]開始的最長回文，找到以s[0]開始的最長回文，將剩下的部分倒序補在字符串前面就是答案。在找以s[0]開始的回文的時候，可以從后向前枚舉回文結束的字符，在第一個找到的地方退出即可。最壞情況下時間復雜度為O(n^2)，用C語言提交AC，704ms。

bool isPalindrom(char* s, int start, int end) {
    while (start < end) {
        if (s[start] != s[end]) return false;
        ++start; --end;
    }
    return true;
}

char* shortestPalindrome(char* s) {
    int pos = strlen(s) - 1;
    for (; pos > 0; --pos) {
        if (s[pos] == s[0] && isPalindrom(s, 0, pos)) break;
    }
    char* res = (char*) malloc(2 * strlen(s) - pos);
    int idx = 0;
    for (int i = strlen(s) - 1; i > pos; --i) res[idx++] = s[i];
    strcpy(res + idx, s);
    return res;
}

但是用C++提交后超時，可能這並不是最佳的答案吧，想到更好的方法后再回來補上代碼（已經想到了，見后文）。

class Solution {
public:
    bool isPalindrom(string &s, int start, int end) {
        while (start < end) {
            if (s[start] != s[end]) return false;
            ++start; --end;
        }
        return true;
    }
    string shortestPalindrome(string s) {
        int pos = s.length() - 1;
        if (pos == 0) return s;
        for (; pos > 0; --pos) {
            if (s[pos] == s[0] && isPalindrom(s, 0, pos)) break;
        }
        string res;
        for (int i = s.length() - 1; i > pos; --i) res.push_back(s[i]);
        res += s;
        return res;
    }
};

 

求最長回文的Manacher算法時間復雜度為O(n)，可以用在這里，但是在判斷最長回文的時候需要加一個條件，就是只比較那些以s[0]開始的回文，而根據Manacher算法的特性，id == p[id]時回文一定是從s[0]開始的。所以下面也是用C++寫可以AC的代碼，因為時間復雜度是O(n)，所以只用了16ms。關於Manacher算法，可以看我另一篇文章：

http://www.cnblogs.com/easonliu/p/4454213.html

class Solution {
public:
    int longestPalindrom(string s) {
        string s1;
        s1.resize(2 * s.length() + 2);
        int idx = 0;
        s1[idx++] = '$';
        s1[idx++] = '#';
        for (char a : s) {
            s1[idx++] = a;
            s1[idx++] = '#';
        }
        vector<int> p(s1.length(), 0);
        int res = 0;
        for (int id = 0, i = 1; i < s1.length(); ++i) {
            if (i < id + p[id]) p[i] = min(p[2 * id - i], id + p[id] - i);
            else p[i] = 1;
            while (s1[i + p[i]] == s1[i - p[i]]) ++p[i];
            if (id + p[id] < i + p[i]) id = i;
            if (p[i] == i) res = max(res, i);
        }
        return res - 1;
    }
    
    string shortestPalindrome(string s) {
        int pos = longestPalindrom(s);
        string res;
        for (int i = s.length() - 1; i >= pos; --i) res.push_back(s[i]);
        return res + s;
    }
};