LeetCode 940 不同的子序列2

最新推荐文章于 2022-10-14 12:03:08 发布

枫流仁武

最新推荐文章于 2022-10-14 12:03:08 发布

阅读量111

点赞数

分类专栏： Leetcode 动态规划

本文链接：https://blog.csdn.net/weixin_42379312/article/details/113976665

版权

动态规划字符串处理子序列算法 Python

关键词由CSDN通过智能技术生成

Leetcode 同时被 2 个专栏收录

154 篇文章 0 订阅

订阅专栏

动态规划

36 篇文章 0 订阅

订阅专栏

给定一个字符串 S，计算 S 的不同非空子序列的个数。

因为结果可能很大，所以返回答案模 10^9 + 7.

示例 1：

输入："abc"
输出：7
解释：7 个不同的子序列分别是 "a", "b", "c", "ab", "ac", "bc", 以及 "abc"。
示例 2：

输入："aba"
输出：6
解释：6 个不同的子序列分别是 "a", "b", "ab", "ba", "aa" 以及 "aba"。
示例 3：

输入："aaa"
输出：3
解释：3 个不同的子序列分别是 "a", "aa" 以及 "aaa"。

利用动态规划解决，解决方法：官方答案

from typing import *
from collections import defaultdict


class Solution:
    def distinctSubseqII(self, S: str) -> int:
        n = len(S)
        dp = [0] * (n + 1)
        dp[0] = 1
        mod = int(10 ** 9 + 7)
        last = defaultdict(int)
        for i, c in enumerate(S):
            dp[i + 1] = 2 * dp[i]
            if c in last:
                dp[i + 1] -= dp[last[c]]
            last[c] = i
        return (dp[len(S)] - 1) % mod