Leetcode 3543. Maximum Weighted K-Edge Path

  • [Leetcode 3543. Maximum Weighted K-Edge Path](#Leetcode 3543. Maximum Weighted K-Edge Path)
    • [1. 解题思路](#1. 解题思路)
    • [2. 代码实现](#2. 代码实现)

1. 解题思路

这一题思路上就是一个遍历的思路,我们只需要考察每一个节点作为起点时,所有长为 k k k的线段的长度,在符合条件的结果当中选出最大值即可。

需要注意的是,由于中间会有大量的重复操作存在,我们需要使用缓存来优化一下执行效率。

2. 代码实现

给出python代码实现如下:

python 复制代码
class Solution:
    def maxWeight(self, n: int, edges: List[List[int]], k: int, t: int) -> int:
        graph = defaultdict(list)
        for u, v, w in edges:
            graph[u].append((v, w))

        @lru_cache(None)
        def dfs(u, k):
            if k == 0:
                return {0}
            if graph[u] == []:
                return set()
            ans = set()
            for v, w in graph[u]:
                nxt_set = dfs(v, k-1)
                for nxt in nxt_set:
                    if nxt + w < t:
                        ans.add(nxt+w)
            return ans
        
        return max(max(dfs(u, k)) if len(dfs(u, k)) > 0 else -1 for u in range(n))

提交代码评测得到:耗时216ms,占用内存48.2MB。

相关推荐
jtymyxmz8 分钟前
刷题日记0726
leetcode
小新学习屋22 分钟前
《剑指offer》-数据结构篇-树
数据结构·算法·leetcode
恣艺2 小时前
LeetCode 1074:元素和为目标值的子矩阵数量
算法·leetcode·矩阵
技术卷3 小时前
详解力扣高频SQL50题之1084. 销售分析 III【简单】
sql·leetcode·oracle
设计师小聂!4 小时前
力扣---------238. 除自身以外数组的乘积
数据结构·算法·leetcode
技术卷5 小时前
详解力扣高频SQL50题之550. 游戏玩法分析 IV【中等】
sql·mysql·leetcode·oracle
科大饭桶6 小时前
数据结构自学Day15 -- 非比较排序--计数排序
数据结构·算法·leetcode·排序算法·c
剪一朵云爱着6 小时前
力扣二叉树的前序中序后序遍历总结
算法·leetcode·二叉树
技术卷11 小时前
详解力扣高频 SQL 50 题之584. 寻找用户推荐人【入门】
sql·leetcode·oracle
小徐不徐说11 小时前
每日一算:华为-批萨分配问题
数据结构·c++·算法·leetcode·华为·动态规划·后端开发