647. Palindromic Substrings

Given a string, your task is to count how many palindromic substrings in this string.

The substrings with different start indexes or end indexes are counted as different substrings even they consist of same characters.

Example 1:

Input: "abc"
Output: 3
Explanation: Three palindromic strings: "a", "b", "c".

 

Example 2:

Input: "aaa"
Output: 6
Explanation: Six palindromic strings: "a", "a", "a", "aa", "aa", "aaa".

 

Note:

  1. The input string length won't exceed 1000.

 

Rust Solution

struct Solution;

impl Solution {
    fn count_substrings(s: String) -> i32 {
        let n = s.len();
        let s: Vec<char> = s.chars().collect();
        let mut res = 0;
        let mut dp = vec![vec![false; n + 1]; n + 1];
        for i in (0..n).rev() {
            for j in i..n {
                if j == i
                    || j == i + 1 && s[i] == s[j]
                    || j > i + 1 && s[i] == s[j] && dp[i + 1][j - 1]
                {
                    dp[i][j] = true;
                    res += 1;
                }
            }
        }
        res
    }
}

#[test]
fn test() {
    let s = "abc".to_string();
    let res = 3;
    assert_eq!(Solution::count_substrings(s), res);
    let s = "aaa".to_string();
    let res = 6;
    assert_eq!(Solution::count_substrings(s), res);
}

Having problems with this solution? Click here to submit an issue on github.