LeetCode题解(1692)：计算分配糖果的不同方式(Python)

长行

于 2021-06-13 20:08:53 发布

阅读量623

点赞数

分类专栏：刷题记录文章标签： leetcode python 动态规划

本文链接：https://blog.csdn.net/Changxing_J/article/details/117264644

版权

刷题记录专栏收录该内容

1966 篇文章

订阅专栏

题目：原题链接（困难）

标签：动态规划

解法	时间复杂度	空间复杂度	执行用时
Ans 1 (Python)	$O (N K)$	$O (N K)$	3620ms (57.89%)
Ans 2 (Python)
Ans 3 (Python)

解法一：

MOD = 10 ** 9 + 7

class Solution:
    def waysToDistribute(self, n: int, k: int) -> int:
        # dp[i][j] 将i个糖果放进j个袋子的方案数
        dp = [[0] * (k + 1) for _ in range(n + 1)]

        for j in range(1, k + 1):
            dp[j][j] = 1
            for i in range(j + 1, n + 1):
                dp[i][j] = (dp[i - 1][j] * j + dp[i - 1][j - 1]) % MOD

        return dp[-1][-1]