Leetcode 3373. Maximize the Number of Target Nodes After Connecting Trees II

  • [Leetcode 3373. Maximize the Number of Target Nodes After Connecting Trees II](#Leetcode 3373. Maximize the Number of Target Nodes After Connecting Trees II)
    • [1. 接替思路](#1. 接替思路)
    • [2. 代码实现](#2. 代码实现)

1. 接替思路

这一题和前一题Leetcode 3372其实整体思路上并没有啥太大的区别,都是考察两棵树上各自每一个节点作为根节点时满足条件的节点个数,然后相加。

唯一的区别在于,题目Leetcode 3372要求的是到根节点深度不超过 k k k,而这道题要求的是深度为偶数的节点个数,两者大差不差,都是用一个遍历即可。

但是这里由于只需要考虑奇偶性,因此我们可以用缓存来减小重复计算,优化整体的执行效率。

2. 代码实现

给出python代码实现如下:

python 复制代码
class Solution:
    def maxTargetNodes(self, edges1: List[List[int]], edges2: List[List[int]]) -> List[int]:
        graph1 = defaultdict(list)
        for u, v in edges1:
            graph1[u].append(v)
            graph1[v].append(u)

        graph2 = defaultdict(list)
        for u, v in edges2:
            graph2[u].append(v)
            graph2[v].append(u)
            
        n, m = len(edges1) + 1, len(edges2) + 1
        
        @lru_cache(None)
        def dfs1(u, p, is_valid):
            if is_valid:
                if (p != -1 and len(graph1[u]) == 1) or (p == -1 and len(graph1[u]) == 0):
                    return 1
                else:
                    return 1 + sum(dfs1(v, u, False) for v in graph1[u] if v != p) 
            else:
                if (p != -1 and len(graph1[u]) == 1) or (p == -1 and len(graph1[u]) == 0):
                    return 0
                else:
                    return sum(dfs1(v, u, True) for v in graph1[u] if v != p) 
        
        @lru_cache(None)
        def dfs2(u, p, is_valid):
            if is_valid:
                if (p != -1 and len(graph2[u]) == 1) or (p == -1 and len(graph2[u]) == 0):
                    return 1
                else:
                    return 1 + sum(dfs2(v, u, False) for v in graph2[u] if v != p) 
            else:
                if (p != -1 and len(graph2[u]) == 1) or (p == -1 and len(graph2[u]) == 0):
                    return 0
                else:
                    return sum(dfs2(v, u, True) for v in graph2[u] if v != p) 
        
        s1 = [dfs1(u, -1, True) for u in range(n)]
        s2 = [dfs2(u, -1, False) for u in range(m)]
        s = max(s2)
        return [x+s for x in s1] 

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

相关推荐
绵绵细雨中的乡音5 小时前
动态规划-第六篇
算法·动态规划
程序员黄同学5 小时前
动态规划,如何应用动态规划解决实际问题?
算法·动态规划
脑子慢且灵7 小时前
蓝桥杯冲刺:一维前缀和
算法·leetcode·职场和发展·蓝桥杯·动态规划·一维前缀和
小王努力学编程9 小时前
动态规划学习——回文子串系列问题【C++】
c++·学习·算法·leetcode·动态规划
Espresso Macchiato18 小时前
Leetcode 3500. Minimum Cost to Divide Array Into Subarrays
leetcode·动态规划·leetcode hard·leetcode 3500·leetcode双周赛153
工一木子19 小时前
大厂算法面试 7 天冲刺:第5天- 递归与动态规划深度解析 - 高频面试算法 & Java 实战
算法·面试·动态规划
pen-ai1 天前
【NLP】15. NLP推理方法详解 --- 动态规划:序列标注,语法解析,共同指代
人工智能·自然语言处理·动态规划
烁3471 天前
每日一题(小白)动态规划篇2
算法·动态规划
Dream it possible!1 天前
LeetCode 热题 100_打家劫舍(83_198_中等_C++)(动态规划)
c++·算法·leetcode·动态规划
烁3471 天前
每日一题(小白)动态规划篇5
算法·动态规划