【Lintcode】1856. Sub-palindrome

記錄演算法發表於2020-12-01

題目地址:

https://www.lintcode.com/problem/sub-palindrome/description

給定一個字串 s s s,求其有多少個不同的迴文子串。

直接列舉中心,然後列舉所有以當前位置為中心的迴文串,加入一個雜湊表中,最後返回雜湊表的size即可。程式碼如下:

import java.util.HashSet;
import java.util.Set;

public class Solution {
    /**
     * @param s: the string
     * @return: the number of substring
     */
    public int countSubstrings(String s) {
        // Write your code here.
        Set<String> set = new HashSet<>();
        // 列舉左端點
        for (int i = 0; i < s.length(); i++) {
            int l = i;
            // 列舉右端點
            for (int j = 0; j < 2; j++) {
                int r = l + j;
                while (l >= 0 && r < s.length() && s.charAt(l) == s.charAt(r)) {
                    set.add(s.substring(l, r + 1));
                    l--;
                    r++;
                }
                
                // 最後要把左端點恢復原狀
                l = i;
            }
        }
        
        return set.size();
    }
}

時間複雜度 O ( l s 2 ) O(l_s^2) O(ls2),空間 O ( l s 3 ) O(l_s^3) O(ls3)