力扣647 回文子串-马拉车算法

647. 回文子串

给你一个字符串 s ，请你统计并返回这个字符串中 回文子串 的数目。

回文字符串 是正着读和倒过来读一样的字符串。

子字符串 是字符串中的由连续字符组成的一个序列。

具有不同开始位置或结束位置的子串，即使是由相同的字符组成，也会被视作不同的子串。

示例 1：

输入：s = “abc”
输出：3
解释：三个回文子串: “a”, “b”, “c”
示例 2：

输入：s = “aaa”
输出：6
解释：6个回文子串: “a”, “a”, “a”, “aa”, “aa”, “aaa”

提示：

1 <= s.length <= 1000
s 由小写英文字母组成

思路：马拉车算法，首先对字符串进行填充改变 前面加"$#“，每个字符中间加一个”#“，新字符串末端加”!"，定义一个f数组来维护不同中心的回文子串的最大半径，维护一个回文串最大区间[l, r]，然后从1开始枚举新字符串，如果当前中心在前面的区间内，分两种情况，一是半径不超过r，此时f[i]可以取关于[l,r]中心点对称的左边的f[l + r - i]，而是半径超过r，就先取到r的距离为f[i]，超过的部分再后面暴力枚举。如果当前中心i在区间外，那就直接定f[i]为1，暴力枚举更新f[i]。更新完f[i]后就得到新的回文串，如果超过[l,r]，就得更新[l,r]，最后就是以i为中心的回文串，最多贡献f[i] / 2个回文子串

class Solution {
    public int countSubstrings(String s) {
        StringBuffer t = new StringBuffer("$#");
        for(int i = 0; i < s.length(); i++){
            t.append(s.charAt(i));
            t.append("#");
        }
        int n = t.length();
        t.append("!");
        int res = 0;
        int[] f = new int[n]; //保存每一个位置回文子串最大的半径
        int l = 0, r = 0;
        for(int i = 1; i < n; i++){
            // 如果i比最右的回文子串还靠右，就开始定义回文子串半径为1，开始暴力枚举
            //如果i在维护的[l, r]回文串内部，f[i]取最小为f[i在回文串左边对应点]或者i到回文串边界的距离，超过的部分再暴力枚举
            f[i] = (i > r) ? 1 : Math.min(f[l + r - i], r - i + 1);
            //确定完f[i]最小值后，开始向两侧寻找最大的回文半径
            while(t.charAt(i + f[i]) == t.charAt(i - f[i]))
                f[i]++;
            
            //如果i为中心的回文串大于r也就是更靠右，就更新l，r
            if(i + f[i] - r + 1 > 0){
                l = i - f[i] + 1;
                r = i + f[i] - 1;
            }
            //以i为中心的回文子串，最多贡献f[i] / 2个回文串
            res += f[i] / 2;
        }
        return res;
    }
}