题意: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加入一个不同与子串的字符“#”恍然大悟。
代码如下:
class Solution {
public:
string shortestPalindrome(string s) {
int sLen = s.size();
string inverStr(s.rbegin(), s.rend());
string tmpStr = s+"#"+inverStr;
vector<int> pi(tmpStr.size()+1, 0);
//pi[i]代表字符串前i个字符中即是前缀又是后缀的最长子串长度
int k = 0;
for(int i = 1;i < tmpStr.size();i++) {
while(k != 0 && tmpStr[k] != tmpStr[i]) {
k = pi[k];
}
if(tmpStr[k] == tmpStr[i]) {
k++;
}
pi[i+1] = k;
}
return inverStr.substr(0, sLen-k)+s;
}
};