Leetcode-LCP 07. 传递信息

最新推荐文章于 2022-06-14 11:41:06 发布

Sherlockkkkz

最新推荐文章于 2022-06-14 11:41:06 发布

阅读量373

点赞数

分类专栏： Leetcode 文章标签： leetcode 算法 python

本文链接：https://blog.csdn.net/weixin_44166338/article/details/123879521

版权

动态规划深度优先递推问题代码实现算法

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

Leetcode 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

在这里插入图片描述

这题可以用两种思想来做，一种是DP(动态规划)，另一种是DFS（深度优先）。

方法一：DP（动态规划）

dp[ i ][ j ] 代表传到第 i 轮传到 j 有几种方式。 所以dp数组的大小应该为 （k+1)×n (还有个第0轮)。
又因为relation数组中包含的是数组，所以应该遍历的是relation中的数组设为 r ，relation是个图，画出来如下图：
在这里插入图片描述
[ 0, 2 ], [ 2, 1 ] 很明显，从 0 到达 1 要经过 2；
所以递推式为：dp[ i ][ relation [ r ][ 1 ]] = dp[ i - 1 ][ relation [ r ][ 0]] + 1
初始化数组dp[0][0] = 1
代码如下：

class Solution:      
    def numWays(self, n: int, relation: List[List[int]], k: int) -> int:
        dp = [[0 for i in range(n)]for i in range(k+1)]   ##初始化数组  Tip：里面的循环是列，外面是行
        dp[0][0]=1
        for i in range(1,k+1):
            for r in range(len(relation)):
                dp[i][relation[r][1]]+=dp[i-1][relation[r][0]]
        return dp[k][n-1]