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。

相关推荐
Jessie_waverider2 天前
LeetCode刷题day29——动态规划(完全背包)
算法·leetcode·动态规划
忘梓.2 天前
解锁动态规划的奥秘:从零到精通的创新思维解析(2)
算法·动态规划
苓诣2 天前
分割等和子集
动态规划
aaal12343 天前
代码随想录算法训练营day46|动态规划part12
算法·动态规划
灰阳阳3 天前
01背包:模板题+实战题
java·算法·动态规划·背包问题·01背包
灰阳阳3 天前
完全背包_模板题
java·算法·动态规划·完全背包
sweetheart7-73 天前
LeetCode1143. 最长公共子序列(2024冬季每日一题 36)
算法·动态规划·力扣·dp·最长公共子序列
m0_749317524 天前
偷懒算法第二天
java·学习·算法·蓝桥杯·动态规划
sweetheart7-74 天前
LeetCode5. 最长回文子串(2024冬季每日一题 35)
c++·算法·leetcode·动态规划·力扣
潜意识起点4 天前
【潜意识Java】蓝桥杯算法有关的动态规划求解背包问题
算法·蓝桥杯·动态规划