LeetCode题解(1420)：生成数组(Python)

最新推荐文章于 2024-09-19 19:39:54 发布

长行

最新推荐文章于 2024-09-19 19:39:54 发布

阅读量243

点赞数

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

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

版权

刷题记录专栏收录该内容

1966 篇文章 55 订阅

订阅专栏

该博客探讨了一种动态规划的解决方案，用于计算在限定长度和条件下，数组搜索代价的最小化问题。时间复杂度和空间复杂度均为O(50×50×100)，并给出了具体的Python实现代码。解法中定义了状态矩阵，并初始化所有长度为1的数组搜索代价。通过迭代更新状态矩阵，最终得出总搜索代价。解决方案以类方法形式返回结果。

摘要由CSDN通过智能技术生成

题目：原题链接（困难）

标签：动态规划

解法	时间复杂度	空间复杂度	执行用时
Ans 1 (Python)	$O (50 \times 50 \times 100)$	$O (50 \times 50 \times 100)$	120ms (94.44%)
Ans 2 (Python)
Ans 3 (Python)

解法一：

MOD = 10 ** 9 + 7

# 定义状态矩阵：dp[n][k][m]
MAX_N, MAX_M, MAX_K = 50, 100, 50
DP = [[[0] * (MAX_M + 1) for _ in range(MAX_K + 1)] for _ in range(MAX_N + 1)]

# 所有长度为1的数组的搜索代价都为1
for j in range(1, MAX_M + 1):
    DP[1][1][j] = 1

for i in range(2, MAX_N + 1):
    for s in range(1, min(MAX_K, i) + 1):  # 搜索代价不会超过数组长度
        prefix = 0
        for j in range(1, MAX_M + 1):
            DP[i][s][j] = (DP[i - 1][s][j] * j + prefix) % MOD
            prefix += DP[i - 1][s - 1][j]


class Solution:
    def numOfArrays(self, n: int, m: int, k: int) -> int:
        return sum(DP[n][k][j] for j in range(1, m + 1)) % MOD