美文网首页
【算法题】2484. 统计回文子序列数目

【算法题】2484. 统计回文子序列数目

作者: 程序员小2 | 来源:发表于2023-08-05 09:57 被阅读0次

题目:

给你数字字符串 s ,请你返回 s 中长度为 5 的 回文子序列 数目。由于答案可能很大,请你将答案对 109 + 7 取余 后返回。

提示:

如果一个字符串从前往后和从后往前读相同,那么它是 回文字符串 。
子序列是一个字符串中删除若干个字符后,不改变字符顺序,剩余字符构成的字符串。

示例 1:

输入:s = "103301"
输出:2
解释:
总共有 6 长度为 5 的子序列:"10330" ,"10331" ,"10301" ,"10301" ,"13301" ,"03301" 。
它们中有两个(都是 "10301")是回文的。
示例 2:

输入:s = "0000000"
输出:21
解释:所有 21 个长度为 5 的子序列都是 "00000" ,都是回文的。
示例 3:

输入:s = "9999900000"
输出:2
解释:仅有的两个回文子序列是 "99999" 和 "00000" 。

提示:

1 <= s.length <= 10^4
s 只包含数字字符。

Java代码

class Solution {
   private static final long MOD = (long) 1e9 + 7;

   public int countPalindromes(String S) {
       var s = S.toCharArray();
       int[] pre = new int[10], suf = new int[10];
       int[][] pre2 = new int[10][10], suf2 = new int[10][10];
       for (var i = s.length - 1; i >= 0; --i) {
           var d = s[i] - '0';
           for (var j = 0; j < 10; ++j)
               suf2[d][j] += suf[j];
           ++suf[d];
       }

       var ans = 0L;
       for (var d : s) {
           d -= '0';
           --suf[d];
           for (var j = 0; j < 10; ++j)
               suf2[d][j] -= suf[j]; // 撤销
           for (var j = 0; j < 10; ++j)
               for (var k = 0; k < 10; ++k)
                   ans += (long) pre2[j][k] * suf2[j][k]; // 枚举所有字符组合
           for (var j = 0; j < 10; ++j)
               pre2[d][j] += pre[j];
           ++pre[d];
       }
       return (int) (ans % MOD);
   }
}

相关文章

网友评论

      本文标题:【算法题】2484. 统计回文子序列数目

      本文链接:https://www.haomeiwen.com/subject/hmvbpdtx.html