Java 实现求最长回文子串

下面以字符串12212321为例，经过上一步，变成了 S[] = "$#1#2#2#1#2#3#2#1#";

然后用一个数组 P[i] 来记录以字符S[i]为中心的最长回文子串向左/右扩张的长度（包括S[i]，也就是把该回文串“对折”以后的长度），比如S和P的对应关系：

S  #  1  #  2  #  2  #  1  #  2  #  3  #  2  #  1  #
P  1  2  1  2  5  2  1  4  1  2  1  6  1  2  1  2  1
(p.s. 可以看出，P[i]-1正好是原字符串中回文串的总长度）

那么怎么计算P[i]呢？该算法增加两个辅助变量（其实一个就够了，两个更清晰）id和mx，其中 id 为已知的 {右边界最大} 的回文子串的中心，mx则为id+P[id]，也就是这个子串的右边界。

然后可以得到一个非常神奇的结论，这个算法的关键点就在这里了：如果mx > i，那么P[i] >= MIN(P[2 * id - i], mx - i)。就是这个串卡了我非常久。实际上如果把它写得复杂一点，理解起来会简单很多：

//记j = 2 * id - i，也就是说 j 是 i 关于 id 的对称点(j = id + (id - i))
if (mx - i > P[j]) 
    P[i] = P[j];
else /* P[j] >= mx - i */
    P[i] = mx - i; // P[i] >= mx - i，取最小值，之后再匹配更新。

当然光看代码还是不够清晰，还是借助图来理解比较容易。

当 mx - i > P[j] 的时候，以S[j]为中心的回文子串包含在以S[id]为中心的回文子串中，由于 i 和 j 对称，以S[i]为中心的回文子串必然包含在以S[id]为中心的回文子串中，所以必有 P[i] = P[j]，见下图。


当 P[j] >= mx - i 的时候，以S[j]为中心的回文子串不一定完全包含于以S[id]为中心的回文子串中，但是基于对称性可知，下图中两个绿框所包围的部分是相同的，也就是说以S[i]为中心的回文子串，其向右至少会扩张到mx的位置，也就是说 P[i] >= mx - i。至于mx之后的部分是否对称，就只能老老实实去匹配了。


对于 mx <= i 的情况，无法对 P[i]做更多的假设，只能P[i] = 1，然后再去匹配了。

于是代码如下：

//输入，并处理得到字符串s
int p[1000], mx = 0, id = 0;
memset(p, 0,  sizeof(p));
for (i = 1; s[i] != '\0'; i++) {
    p[i] = mx > i ? min(p[2*id-i], mx-i) : 1;
     while (s[i + p[i]] == s[i - p[i]]) p[i]++;
     if (i + p[i] > mx) {
        mx = i + p[i];
        id = i;
    }
}
//找出p[i]中最大的

Java 代码如下：

package Exercises;


import java.io.FileNotFoundException;
import java.util.Scanner;


public class Palindrome {


public static void main(String[] args) throws FileNotFoundException{
// TODO Auto-generated method stub
 int caseNum = 0;
     Scanner sc = new Scanner(System.in);
       while (true) {
           String str = sc.nextLine();
           if (str.equals("END")) {
               break;
           } else {
               caseNum ++;
               System.out.println("Case " + caseNum + ": " + getLongestPalindrome(str));
           }
       }
}
 public static int getLongestPalindrome(String A) {
       // write code here
            // 1.构造新的字符串
       // 为了避免奇数回文和偶数回文的不同处理问题，在原字符串中插入'#'，将所有回文变成奇数回文
       StringBuilder newStr = new StringBuilder();
       newStr.append('#');
       for (int i = 0; i < A.length(); i ++) {
           newStr.append(A.charAt(i));
           newStr.append('#');
       }
       // p[i]表示以i为中心的回文的最大半径，i至少为1，即该字符本身
       int [] p = new int[newStr.length()];
       // mx表示已知的回文中，最右的边界的坐标
       int mx = -1;
       // id表示已知的回文中，拥有最右边界的回文的中点坐标
       int id = -1;
       // 2.计算所有的p
       // 这个算法是O(n)的，因为mx只会随着里层while的迭代而增长，不会减少。
       for (int i = 0; i < newStr.length(); i ++) {
           // 2.1.确定一个最小的半径
           int r = 1;
           if (i <= mx) {
               r = Math.min(p[id] - i + id, p[2 * id - i]);
           }
           // 2.2.尝试更大的半径
           while (i - r >= 0 && i + r < newStr.length() && newStr.charAt(i - r) == newStr.charAt(i + r)) {
               r++;
           }
           // 2.3.更新边界和回文中心坐标
           if (i + r - 1> mx) {
               mx = i + r - 1;
               id = i;
           }
           p[i] = r;
       }
       
       // 3.扫描一遍p数组，找出最大的半径
       int maxLength = 0;
       for (int r : p) {
           if (r > maxLength) {
               maxLength = r;
           }
       }
       return maxLength - 1;
   }
}