LeetCode 5 Longest Palindromic Substring(最长回文字串)

1
2
3

Input: "babad"
Output: "bab"
Note: "aba" is also a valid answer.

1
2

Input: "cbbd"
Output: "bb"

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

class Solution
{

public:
    string longestPalindrome(string s)
    {
        // 马拉车算法
        if (s.empty())
            return "";

        // 先处理一下字符串，给每个字符之间加入#，来解决对称位置的奇偶问题
        // 保证所有的串和回文串都是奇数长度    
        string prep = "#";
        for (auto ch : s)
        {
            prep += ch;
            prep += "#";
        }

        const int size_p = prep.size();
        // dp(i) 表示以 i 为中心的最长回文的半径
        vector<int> dp(size_p, 0);
    
        // bCur 是以 center 为中心，最长回文的右边界
        int center = 0, bCur = 0;
        for (int i = 0; i < size_p; i++)
        {
            // 遍历是从左至右，所以
            // center一定小于等于i
            // 到i的时候，dp[mirror]是最长回文半径
            // mirror 是以 center 为中心，i的对称点
            int mirror = center - (i - center);

            // 这里分三种情况：
            // 1. dp[mirror] > bCur - i,那么 dp[i] = bCur - i，否则的话以 center 为中心，到 bCur 的长度就不是最长回文半径了，与假设不符
            // 2. dp[mirror] < bCur - i,那么 dp[i] = dp[mirror],根据 i 和 mirror 是以 center 为中心对称，且 dp[mirror] 在 center 为中心的回文之内，如果 dp[i] 比 dp[mirror] 更大的话，那么与 dp[mirror] 是最大回文前置条件不符
            // 3. dp[mirror] == bCur - i，那么 i 的最长回文至少是包含 mirror 的，即 dp[i] >= dp[mirror]，所以还需要后面继续扩展 i 的回文长度的处理。
            dp[i] = bCur <= i ? 0 : min(bCur - i, dp[mirror]);

            // 继续扩展 i 的回文长度，只有第三种情况的时候才有效，前两种情况必不满足 prep[start - 1] == prep[end + 1]
            int start = i - dp[i], end = i + dp[i];
            while (start - 1 >= 0 && end + 1 < size_p && prep[start - 1] == prep[end + 1])
            {
                --start;
                ++end;
                ++dp[i];
            }

            // 更新 center 和 bCur
            if (i + dp[i] > bCur)
            {
                bCur = i + dp[i];
                center = i;
            }
        }
        center = max_element(dp.begin(), dp.end()) - dp.begin();
        return s.substr((center - dp[center]) / 2, dp[center]);
    }
};