LeetCode之Shortest Palindrome

/*为了获取最短的回文串，需要找到字符串从头开始的最长回文子串，然后取剩下部分的字符串的逆序加到字符串的前头，即可获得最短回文子串。所以解决该问题的关键部分是寻找以第一个字符开始的最长回文子串，有以下几种方法：1.采用暴力查找，大数据下会超时。2.借用KMP算法来寻找最长回文子串，即本代码中所用方法；3.采用Manacher算法。*/class Solution {public:string shortestPalindrome(string s) {if(s.size() <= 1) return s;int idx = kmp_search(s);int i = idx, j = idx+1;while(i >= 0 && j < s.size() && s[i] == s[j]){--i;++j;}if(i >= 0){i = idx - 1;j = idx + 1;while(i >= 0 && j < s.size() && s[i] == s[j]){--i;++j;};}if(j >= s.size()) return s;string sub = s.substr(j, s.size()-j);reverse(sub.begin(), sub.end());return sub + s;}int kmp_search(const string &s){vector<int> next;get_next(s, next);int i = 0, j = s.size() - 1;while(i < j){if(i == -1 || s[i] == s[j]){++i;--j;}else i = next[i];}return j;}void get_next(const string &s, vector<int> &next){next.assign(s.size(), 0);int i = 0, j = -1;next[i] = j;while(i < s.size()-1){if(j == -1 || s[i] == s[j]){++i;++j;next[i] = s[i] == s[j] ? next[j] : j;}else j = next[j];}}};