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。

相关推荐
明月清了个风11 分钟前
数据结构与算法学习笔记(Acwing提高课)----动态规划·数字三角形
笔记·学习·算法·动态规划
小指纹11 小时前
动态规划(一)【背包】
c++·算法·动态规划
寂空_21 小时前
【算法笔记】动态规划基础(一):dp思想、基础线性dp
c++·笔记·算法·动态规划
庐阳寒月2 天前
动态规划算法:完全背包类问题
数据结构·c++·算法·动态规划
Dream it possible!2 天前
LeetCode 热题 100_分割等和子集(89_416_中等_C++)(动态规划)
c++·leetcode·动态规划
Aqua Cheng.2 天前
代码随想录第三十七天|华为秋季笔试真题230823
java·算法·leetcode·华为·动态规划
_GR2 天前
2025年蓝桥杯第十六届C&C++大学B组真题及代码
c语言·数据结构·c++·算法·贪心算法·蓝桥杯·动态规划
不是仙人的闲人3 天前
算法之动态规划
数据结构·c++·算法·动态规划
callJJ3 天前
Floyd算法求解最短路径问题——从零开始的图论讲解(3)
java·算法·动态规划·图论·dijkstra算法·floyd算法·最短路径问题
谦川4 天前
蓝桥杯 蜗牛 动态规划
职场和发展·蓝桥杯·动态规划