LCP 07. 传递信息

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

Stig.Huang

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

阅读量170

点赞数 1

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

本文链接：https://blog.csdn.net/weixin_50791900/article/details/109805054

版权

python 同时被 2 个专栏收录

115 篇文章 9 订阅

订阅专栏

leetcode

69 篇文章 1 订阅

订阅专栏

本题可以运用dfs加递归的方法解决：
首先定义一个函数，分别将列表、开始的点和走的步数
先设置终止条件，则是当走的步数为k时，如果路径最后输出的是符合题目条件的n-1，则将本方法记录在结果ans中
同时在最后返回0（停止本次搜索）
之后将列表中的每一个0分别设置为start，重复递归
因为数组中多个0都可以成为开始的起点
所以将多个0重复为最最开始的起点
最后返回记录了每种方法的列表ans的长度
则是共有几种传递方法

class Solution:
    def numWays(self, n: int, relation: List[List[int]], k: int) -> int:
        ans = []

        def dfs(relation, start, step):
 
            if step == k: 
                if start == n-1:  
                    ans.append(1)  
                return 0
            
            for i in range(len(relation)):
                if relation[i][0] == start:
                    dfs(relation, relation[i][1], step+1)
                    
        for i in range(len(relation)): 
            if relation[i][0] == 0:
                dfs(relation, relation[i][1], 1) 
        
        return len(ans)