g2401_2500.s2484_count_palindromic_subsequences.Solution Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of leetcode-in-java17 Show documentation

Java Solution for LeetCode algorithm problems, continually updating

There is a newer version: 1.24

package g2401_2500.s2484_count_palindromic_subsequences;

// #Hard #String #Dynamic_Programming #2023_01_26_Time_93_ms_(76.16%)_Space_40.2_MB_(99.74%)

public class Solution {
    public int countPalindromes(String s) {
        int n = s.length();
        long ans = 0;
        int mod = (int) 1e9 + 7;
        int[] count = new int[10];
        for (int i = 0; i < n; i++) {
            long m = 0;
            for (int j = n - 1; j > i; j--) {
                if (s.charAt(i) == s.charAt(j)) {
                    ans += (m * (j - i - 1));
                    ans = ans % mod;
                }
                m += count[s.charAt(j) - '0'];
            }
            count[s.charAt(i) - '0']++;
        }
        return (int) ans;
    }
}