【To Read】Shortest Palindrome(KMP)
题意: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”.
题解:首先想到一种可行的方案是将原串str翻转成inverStr直接放在前面显然构成回文,则我们试着将inverStr往右移,也就是减少添加字符个数,若翻转串与原串完全重合,则仍然满足要求,显然最少添加就是使翻转串与原串的最大重合。
更进一步,我们知道重合的部分也是个回文串,即我们也是要求最长的str是回文子串的前缀,可以使用二维DP求解(TLE)。
其实我们注意到重合的部分是inverStr的后缀,则我们可以考虑使用KMP算法,因为它就是用来求解即是前缀又是后缀的最长子串。
一开始遇到瓶颈,因为直接将str+inverStr,求解后的pi[N]可能大于str的长度,则这个解说明不了问题。
看到discuss加入一个不同与子串的字符“#”恍然大悟。
代码如下:
1 class Solution { 2 public: 3 string shortestPalindrome(string s) { 4 int sLen = s.size(); 5 string inverStr(s.rbegin(), s.rend()); 6 string tmpStr = s+"#"+inverStr; 7 8 vector<int> pi(tmpStr.size()+1, 0); 9 //pi[i]代表字符串前i个字符中即是前缀又是后缀的最长子串长度 10 int k = 0; 11 for(int i = 1;i < tmpStr.size();i++) { 12 while(k != 0 && tmpStr[k] != tmpStr[i]) { 13 k = pi[k]; 14 } 15 if(tmpStr[k] == tmpStr[i]) { 16 k++; 17 } 18 pi[i+1] = k; 19 } 20 21 return inverStr.substr(0, sLen-k)+s; 22 } 23 };