题目描述(困难难度)

214*. Shortest Palindrome - 图1

在字符串开头补充最少的字符,使得当前字符串成为回文串。

解法一 暴力

先判断整个字符串是不是回文串,如果是的话,就直接将当前字符串返回。不是的话,进行下一步。

判断去掉末尾 1 个字符的字符串是不是回文串,如果是的话,就将末尾的 1 个字符加到原字符串的头部返回。不是的话,进行下一步。

判断去掉末尾 2 个字符的字符串是不是回文串,如果是的话,就将末尾的 2 个字符倒置后加到原字符串的头部返回。不是的话,进行下一步。

判断去掉末尾 3 个字符的字符串是不是回文串,如果是的话,就将末尾的 3 个字符倒置后加到原字符串的头部返回。不是的话,进行下一步。

直到判断去掉末尾的 n - 1 个字符,整个字符串剩下一个字符,把末尾的 n - 1 个字符倒置后加到原字符串的头部返回。

举个例子,比如字符串 abbacd

  1. 原字符串 abbacd
  2. 先判断 abbacd 是不是回文串, 发现不是, 执行下一步
  3. 判断 abbac 是不是回文串, 发现不是, 执行下一步
  4. 判断 abba 是不是回文串, 发现是,将末尾的 2 个字符 cd 倒置后加到原字符串的头部,
  5. dcabbacd

代码的话,判断是否是回文串的话可以用 125 题 的思想,利用双指针法。

  1. //判断是否是回文串, 传入字符串的范围
  2. public boolean isPalindromic(String s, int start, int end) {
  3. char[] c = s.toCharArray();
  4. while (start < end) {
  5. if (c[start] != c[end]) {
  6. return false;
  7. }
  8. start++;
  9. end--;
  10. }
  11. return true;
  12. }
  13. public String shortestPalindrome(String s) {
  14. int end = s.length() - 1;
  15. //找到回文串的结尾, 用 end 标记
  16. for (; end > 0; end--) {
  17. if (isPalindromic(s, 0, end)) {
  18. break;
  19. }
  20. }
  21. //将末尾的几个倒置然后加到原字符串开头
  22. return new StringBuilder(s.substring(end + 1)).reverse() + s;
  23. }

遗憾的是超时了(前几天这个方法还能是过的,今天突然就超时了,官方应该是增加了 case)。

214*. Shortest Palindrome - 图2

上边字符串的长度达到了四万多,类似于这样的 aaaaaaaaaaaaaacdaaaaaaaaaaaaaa,每次调用判断字符串是否是回文串的时候,都需要判断很多次,最终时间复杂度达到了 O(n²),造成了超时。

解法二

参考 这里

根据解法一,我们其实就是在寻找从开头开始的最长回文串(这个很关键,后边所有的解法都是基于这个了),然后将末尾的除去最长回文串部分的几个字符倒置后加到原字符串开头即可。

我们只需要两个指针, iji 初始化为 0j 初始化为字符串长度减 1。然后依次判断 s[i]s[j] 是否相同,相同的话, i 就进行加 1j 进行减 1s[i]s[j] 不同的话,只将 j 进行减 1

看几个例子。

  1. abbacde
  2. a b b a c d e
  3. ^ ^
  4. i j
  5. 如上所示, s[i] != s[j], j--
  6. a b b a c d e
  7. ^ ^
  8. i j
  9. 如上所示, s[i] != s[j], j--
  10. a b b a c d e
  11. ^ ^
  12. i j
  13. 如上所示, s[i] != s[j], j--
  14. a b b a c d e
  15. ^ ^
  16. i j
  17. 如上所示, s[i] == s[j], i++, j--
  18. a b b a c d e
  19. ^ ^
  20. i j
  21. 如上所示, s[i] == s[j], i++, j--
  22. a b b a c d e
  23. ^ ^
  24. j i
  25. 如上所示, s[i] == s[j], i++, j--
  26. a b b a c d e
  27. ^ ^
  28. j i
  29. 如上所示, s[i] == s[j], i++, j--
  30. a b b a c d e
  31. ^ ^
  32. j i
  33. 如上所示, j < 0, 结束循环。
  34. 此时 i 指向最长回文串的下一个字符串,我们只需要把 i 最后的字符倒置加到开头即可。

当然,上边是最理想的情况,如果 j 在最长回文串外提前出现了和 i 相同的字符会有影响吗?

  1. abbacba
  2. a b b a c b a
  3. ^ ^
  4. i j
  5. 如上所示, s[i] == s[j], i++, j--
  6. a b b a c b a
  7. ^ ^
  8. i j
  9. 如上所示, s[i] == s[j], i++, j--
  10. a b b a c b a
  11. ^ ^
  12. i j
  13. 如上所示, s[i] != s[j], j--
  14. a b b a c b a
  15. ^ ^
  16. i j
  17. 如上所示, s[i] != s[j], j--
  18. a b b a c b a
  19. ^
  20. i
  21. j
  22. 如上所示, s[i] == s[j], i++, j--
  23. a b b a c b a
  24. ^ ^
  25. j i
  26. 如上所示, s[i] != s[j], j--
  27. a b b a c b a
  28. ^ ^
  29. j i
  30. 如上所示, s[i] == s[j], i++, j--
  31. a b b a c d e
  32. ^ ^
  33. j i
  34. 如上所示, j < 0, 结束循环。
  35. 会发现此时 i 和之前一样, 依旧指向最长回文串的下一个字符,我们只需要把 i 到最后的字符倒置加到开头即可。

可以看到上边的两种情况,只要 j 进入了最长回文子串,一定会使得 i 走出最长回文子串。所以我们可以利用双指针写一下代码了。

  1. public String shortestPalindrome(String s) {
  2. int i = 0, j = s.length() - 1;
  3. char[] c = s.toCharArray();
  4. while (j >= 0) {
  5. if (i == j){
  6. continue;
  7. }
  8. if (c[i] == c[j]) {
  9. i++;
  10. }
  11. j--;
  12. }
  13. //此时代表整个字符串是回文串
  14. if (i == s.length()) {
  15. return s;
  16. }
  17. //后缀
  18. String suffix = s.substring(i);
  19. //后缀倒置
  20. String reverse = new StringBuilder(suffix).reverse().toString();
  21. //加到开头
  22. return reverse + s;
  23. }

看起来没什么问题,但还有一种情况,那就是 i 提前走出了最长回文子串,看下边的例子。

  1. ababbcefbbaba
  2. a b a b b c e f b b a b a
  3. ^ ^
  4. i j
  5. i j 同时移动, 一直是相等, 直到下边的情况
  6. a b a b b c e f b b a b a
  7. ^ ^
  8. i j
  9. 然后继续移动, 最后就变成了下边的样子
  10. a b a b b c e f b b a b a
  11. ^ ^
  12. j i
  13. 会发现此时 0 i - 1 并不是一个回文串, 所以我们需要递归的去解决这个问题

此时我们并没有找到最长回文串,但是我们可以肯定最长回文串一定在 0i 之间,所以我们只需要递归的从s[0, i) 中继续寻找最长回文串即可。

因为上边的所有情况,都保证了 i 一定可以走出最长回文串,只不过可能超出一部分,所以用递归解决即可。代码的整体框架不需要改变。

  1. public String shortestPalindrome(String s) {
  2. int i = 0, j = s.length() - 1;
  3. char[] c = s.toCharArray();
  4. while (j >= 0) {
  5. if (c[i] == c[j]) {
  6. i++;
  7. }
  8. j--;
  9. }
  10. //此时代表整个字符串是回文串
  11. if (i == s.length()) {
  12. return s;
  13. }
  14. //后缀
  15. String suffix = s.substring(i);
  16. //后缀倒置
  17. String reverse = new StringBuilder(suffix).reverse().toString();
  18. //递归 s[0,i),寻找开头开始的最长回文串,将其余部分加到开头和结尾
  19. return reverse + shortestPalindrome(s.substring(0, i)) + suffix;
  20. }

这个解法相对解法一会好一些,但对于某些极端情况,时间复杂度依旧会达到 O(n²)。比如下边的例子。

  1. aababababa
  2. a a b a b a b a b a
  3. ^ ^
  4. i j
  5. 如上所示, s[i] == s[j], i++, j--
  6. a a b a b a b a b a
  7. ^ ^
  8. i j
  9. 如上所示, s[i] != s[j], j--
  10. a a b a b a b a b a
  11. ^ ^
  12. i j
  13. 如上所示, 此时 i j 之间是一个回文串,所以 i j 最终会变成下边的样子
  14. a a b a b a b a b a
  15. ^ ^
  16. j i
  17. 结合上边的代码,接下来去掉末尾字符,将对下边的字符串进行递归
  18. a a b a b a b a
  19. 此时会发现和最开始的结构一样,最终结果是去掉末尾的两个字符,继续对下边的字符串递归
  20. a a b a b a
  21. 此时会发现和最开始的结构一样,最终结果是去掉末尾的两个字符,继续对下边的字符串递归
  22. a a b a
  23. 此时会发现和最开始的结构一样,最终结果是去掉末尾的两个字符,继续对下边的字符串递归
  24. a a
  25. 此时是回文串了,递归结束

所以每次递归只会减少两个字符,递归路径如下

  1. a a b a b a b a b a
  2. a a b a b a b a
  3. a a b a b a
  4. a a b a
  5. a a

如果初始字符串是上边的结构,即 aaba...ba...ba...ba,有几万个 ba 的话,和解法一一样会造成超时。由于 leetcode 中没有这种 case ,所以这个解法也就 AC 了。

解法三

寻找开头开始的最长回文串,我们回到更暴力的方法。

将原始字符串逆序,然后比较对应的子串即可判断是否是回文串。举个例子。

  1. abbacd
  2. s: abbacd, 长度记为 n
  3. r: dcabba, 长度记为 n
  4. 判断 s[0,n) r[0,n)
  5. abbacd != dcabba
  6. 判断 s[0,n - 1) r[1,n)
  7. abbac != cabba
  8. 判断 s[0,n - 2) r[2,n)
  9. abba == abba
  10. 从开头开始的最长回文串也就找到了, 接下来只需要使用之前的方法。
  11. 将末尾不是回文串的部分倒置加到原字符串开头即可。

代码的话,也很好写了。

  1. public String shortestPalindrome(String s) {
  2. String r = new StringBuilder(s).reverse().toString();
  3. int n = s.length();
  4. int i = 0;
  5. for (; i < n; i++) {
  6. if (s.substring(0, n - i).equals(r.substring(i))) {
  7. break;
  8. }
  9. }
  10. return new StringBuilder(s.substring(n - i)).reverse() + s;
  11. }

然后它竟然 AC 了,当然这个时间复杂度是 O(n²),之所以通过了,还是取决于 test cases

解法四

在解法三倒置的基础上进行一下优化,参考 这里-method-using-Rabin-Karp-rolling-hash)。

用到了字符串匹配算法 RK 算法的思想,也就是滚动哈希。

解法三中,每次比较两个字符串是否相等都需要一个字符一个字符比较,如果我们把字符串通过 hash 算法映射到数字,就可以只判断数字是否相等即可。

hash 算法,这里的话,我们将 a 看做 1b 看做 2 … 以此类推,然后把字符串看做是 26 进制的一个数字,将其转为十进制后的值作为 hash 值。

也许需要一些进制转换的知识,可以参考 这里

举个例子,对于 abcd

  1. a b c d
  2. 1 2 3 4
  3. 26^3 26^2 26 1

那么 abcdhash 值就是

214*. Shortest Palindrome - 图3

这样做的好处是,我们可以通过前一个字符串的 hash 值,算出当前字符串的 hash 值。

举个例子。

对于字符串 abb ,如果我们知道了它的 hash 值是 x ,那么对于 abbahash 值,因为新增加的数字 a 对应 1,所以 abbahash 值就是 (x * 26 + 1)

所以代码可以写成下边的样子。

  1. public String shortestPalindrome(String s) {
  2. int n = s.length(), pos = -1;
  3. int b = 26; // 基数
  4. int pow = 1; // 为了方便计算倒置字符串的 hash 值
  5. char[] c = s.toCharArray();
  6. int hash1 = 0, hash2 = 0;
  7. for (int i = 0; i < n; i++, pow = pow * b) {
  8. hash1 = hash1 * b + (c[i] - 'a' + 1);
  9. // 倒置字符串的 hash 值, 新增的字符要放到最高位
  10. hash2 = hash2 + (c[i] - 'a' + 1) * pow;
  11. if (hash1 == hash2) {
  12. pos = i;
  13. }
  14. }
  15. return new StringBuilder(s.substring(pos + 1)).reverse() + s;
  16. }

理论上,上边的代码是可行的,但会发现出现了 wrong answer

214*. Shortest Palindrome - 图4

我猜测下原因,不是十分确定。

最直接的问题肯定是由于我们用 int 存储 hash 值,所以一定会出现溢出的情况。溢出以后,接着带来了 hash 冲突,从而使得相同的 hash 值,但是字符串并不相同。

基于上边的分析,我们可以在 pos = i 之前判断一下当前是否是回文串。

  1. public boolean isPalindromic(String s, int start, int end) {
  2. char[] c = s.toCharArray();
  3. while (start < end) {
  4. if (c[start] != c[end]) {
  5. return false;
  6. }
  7. start++;
  8. end--;
  9. }
  10. return true;
  11. }
  12. public String shortestPalindrome(String s) {
  13. int n = s.length(), pos = -1;
  14. int b = 26; // 基数
  15. int pow = 1; // 为了方便计算倒置字符串的 hash 值
  16. char[] c = s.toCharArray();
  17. int hash1 = 0, hash2 = 0;
  18. for (int i = 0; i < n; i++, pow = pow * b) {
  19. hash1 = hash1 * b + (c[i] - 'a' + 1);
  20. // 倒置字符串的 hash 值, 新增的字符要放到最高位
  21. hash2 = hash2 + (c[i] - 'a' + 1) * pow;
  22. if (hash1 == hash2) {
  23. //确认下当前是否是回文串
  24. if (isPalindromic(s,0,i)) {
  25. pos = i;
  26. }
  27. }
  28. }
  29. return new StringBuilder(s.substring(pos + 1)).reverse() + s;
  30. }

但是超时了。

214*. Shortest Palindrome - 图5

然后就是换 hash 算法,我们可以把每次的结果取模,这样就不会溢出了。

  1. public String shortestPalindrome(String s) {
  2. int n = s.length(), pos = -1;
  3. int b = 26; // 基数
  4. int pow = 1; // 为了方便计算倒置字符串的 hash 值
  5. char[] c = s.toCharArray();
  6. int hash1 = 0, hash2 = 0;
  7. int mod = 1000000;
  8. for (int i = 0; i < n; i++, pow = (pow * b) % mod) {
  9. hash1 = (hash1 * b + (c[i] - 'a' + 1)) % mod;
  10. // 倒置字符串的 hash 值, 新增的字符要放到最高位
  11. hash2 = (hash2 + (c[i] - 'a' + 1) * pow)% mod;
  12. if (hash1 == hash2) {
  13. pos = i;
  14. }
  15. }
  16. return new StringBuilder(s.substring(pos + 1)).reverse() + s;
  17. }

虽然这种方法 AC 了,但我觉得是侥幸的,我觉得即使每次取模,并不能保证不会出现 hash 冲突,只是当前的 test case 没有出现 hash 冲突。当然这是我的想法,并不是很确定,大家有其他想法欢迎和我交流。

感谢 @franklinqin0 指出,上边确认当前是否是回文串的时候,我们调用了 isPalindromic ,但超时了,这里的话我们还可以和它的逆置字符串进行比较。

  1. public String shortestPalindrome(String s) {
  2. int n = s.length(), pos = -1;
  3. int b = 26; // 基数
  4. int pow = 1; // 为了方便计算倒置字符串的 hash 值
  5. char[] c = s.toCharArray();
  6. String rev = new StringBuilder(s).reverse().toString();
  7. int hash1 = 0, hash2 = 0;
  8. for (int i = 0; i < n; i++, pow = pow * b) {
  9. hash1 = hash1 * b + (c[i] - 'a' + 1);
  10. // 倒置字符串的 hash 值, 新增的字符要放到最高位
  11. hash2 = hash2 + (c[i] - 'a' + 1) * pow;
  12. if (hash1 == hash2) {
  13. if (s.substring(0, i + 1).equals(rev.substring(n - i - 1))) {
  14. pos = i;
  15. }
  16. }
  17. }
  18. return new StringBuilder(s.substring(pos + 1)).reverse() + s;
  19. }

这样做的话就不会超时了,但如果分析时间复杂度的话其实是一样的,很神奇。

解法五

参考 这里

这个解法的前提是你熟悉另一种字符串匹配算法,即 KMP 算法。推荐两个链接,大家可以先学习一下,我就不多说了。KMP 算法代码简单,但理解求 next 数组的话,确实有些麻烦。

http://jakeboxer.com/blog/2009/12/13/the-knuth-morris-pratt-algorithm-in-my-own-words/

https://learnku.com/articles/10622/introduction-of-kmp-algorithm-and-derivation-of-next-array

如果熟悉了 KMP 算法,下边就简单了。

再回想一下解法三,倒置字符串的思路,依次比较对应子串。

  1. abbacd
  2. s: abbacd, 长度记为 n
  3. r: dcabba, 长度记为 n
  4. 我们把两个字符串写在一起
  5. abbacd dcabba
  6. 判断 abbacd dcabba 是否相等
  7. 判断 abbac cabba 是否相等
  8. 判断 abba abba 是否相等

如果我们把 abbacd dcabba看成一个字符串,中间加上一个分隔符 #abbacd#dcabba

回味一下上边的三条判断,判断 XXX 和 XXX 是否相等,按列看一下。

左半部分 abbacdabbac , abba 其实就是 abbacd#dcabba 的一些前缀。

右半部分dcabbacabbaabba 其实就是 abbacd#dcabba 的一些后缀。

寻找前缀和后缀相等。

想一想 KMP 算法,这不就是 next 数组做的事情吗。

而我们中间加了分隔符,也就保证了前缀和后缀相等时,前缀一定在 abbacd 中。

换句话说,我们如果求出了 abbacd#dcabbanext 数组,因为我们构造的字符串后缀就是原字符串的倒置,前缀后缀相等时,也就意味着当前前缀是一个回文串,而 next 数组是寻求最长的前缀,我们也就找到了开头开始的最长回文串。

因为 next 数组的含义并不统一,但 KMP 算法本质上都是一样的,所以下边的代码仅供参考。

我的 next 数组 next[i] 所考虑的对应字符串不包含 s[i]

  1. public String shortestPalindrome(String s) {
  2. String ss = s + '#' + new StringBuilder(s).reverse();
  3. int max = getLastNext(ss);
  4. return new StringBuilder(s.substring(max)).reverse() + s;
  5. }
  6. //返回 next 数组的最后一个值
  7. public int getLastNext(String s) {
  8. int n = s.length();
  9. char[] c = s.toCharArray();
  10. int[] next = new int[n + 1];
  11. next[0] = -1;
  12. next[1] = 0;
  13. int k = 0;
  14. int i = 2;
  15. while (i <= n) {
  16. if (k == -1 || c[i - 1] == c[k]) {
  17. next[i] = k + 1;
  18. k++;
  19. i++;
  20. } else {
  21. k = next[k];
  22. }
  23. }
  24. return next[n];
  25. }

解法六

参考 这里-solution-based-on-Manacher’s-algorithm) 。

大家还记得 第 5 题 吗?求最长回文子串。

这里我们已经把题目转换成了求开头开始的最长回文子串,很明显这个问题只是第 5 题的子问题了。但这道题时间复杂度差不多只有 O(n) 才会通过。这就必须使用 第 5 题 介绍的马拉车算法了。

直接把马拉车算法粘贴过来即可,然后在最后稍微修改一下即可。大家不熟悉的话,可以参考 一文让你彻底明白马拉车算法

  1. public String preProcess(String s) {
  2. int n = s.length();
  3. if (n == 0) {
  4. return "^$";
  5. }
  6. String ret = "^";
  7. for (int i = 0; i < n; i++)
  8. ret += "#" + s.charAt(i);
  9. ret += "#$";
  10. return ret;
  11. }
  12. // 马拉车算法
  13. public String shortestPalindrome(String s) {
  14. String T = preProcess(s);
  15. int n = T.length();
  16. int[] P = new int[n];
  17. int C = 0, R = 0;
  18. for (int i = 1; i < n - 1; i++) {
  19. int i_mirror = 2 * C - i;
  20. if (R > i) {
  21. P[i] = Math.min(R - i, P[i_mirror]);// 防止超出 R
  22. } else {
  23. P[i] = 0;// 等于 R 的情况
  24. }
  25. // 碰到之前讲的三种情况时候,需要利用中心扩展法
  26. while (T.charAt(i + 1 + P[i]) == T.charAt(i - 1 - P[i])) {
  27. P[i]++;
  28. }
  29. // 判断是否需要更新 R
  30. if (i + P[i] > R) {
  31. C = i;
  32. R = i + P[i];
  33. }
  34. }
  35. //这里的话需要修改
  36. int maxLen = 0;
  37. int centerIndex = 0;
  38. for (int i = 1; i < n - 1; i++) {
  39. int start = (i - P[i]) / 2;
  40. //我们要判断当前回文串是不是开头是不是从 0 开始的
  41. if (start == 0) {
  42. maxLen = P[i] > maxLen ? P[i] : maxLen;
  43. }
  44. }
  45. return new StringBuilder(s.substring(maxLen)).reverse() + s;
  46. }

这道题太强了,六种解法,各有特色。把这道题捋下来着实不易,涉及到很多算法。但懂了之后确实心旷神怡。

花时间最多的地方其实在 KMP 那里,求 next 数组确实难理解一些。然后递归解法,看起来简单,其实理解起来的话也没那么容易。最后没想到又回到了马拉车算法,不得不再佩服一下这个解法,神仙操作,直接将时间复杂度优化到了 O(n)

windliang wechat

添加好友一起进步~

如果觉得有帮助的话,可以点击 这里 给一个 star 哦 ^^

如果想系统的学习数据结构和算法,强烈推荐一个我之前学过的课程,可以点击 这里 查看详情