小黑周末惊心动魄进“村”赶考，周一继续拖着疲惫的身体来实习的leetcode之旅:968. 监控二叉树

最新推荐文章于 2024-07-25 17:54:26 发布

爱喝喜茶爱吃烤冷面的小黑黑

最新推荐文章于 2024-07-25 17:54:26 发布

阅读量430

点赞数 1

分类专栏：小黑的日常leetcode之旅文章标签： leetcode 算法深度优先

本文链接：https://blog.csdn.net/qq_37418807/article/details/128613086

版权

小黑的日常leetcode之旅专栏收录该内容

225 篇文章 1 订阅

订阅专栏

小黑看完题解思路后代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def minCameraCover(self, root: Optional[TreeNode]) -> int:
        '''
            状态转移:
                0:无覆盖
                1:被覆盖
                2:放置摄像头
        '''
        self.result = 0
        # 定义递归函数
        def dfs(node):
            # 结点为空，当成已覆盖
            if not node:
                return 1
            # 后序遍历，判断孩子结点的顺序
            left = dfs(node.left)
            right = dfs(node.right)
            # 左右结点都被覆盖，则该结点不必覆盖
            if left == right == 1:
                return 0
            # 左右结点全是有相机的情况下,该结点被覆盖
            elif left == 2 and right == 2 :
                return 1
            # 其他情况，需要安装相机
            elif left == 0 or right == 0:
                self.result += 1
                return 2
            # 一个有相机 一个被覆盖，则这个被覆盖
            return 1
        root_status = dfs(root)
        # 如果root没有被覆盖，则安装相机
        if not root_status:
            self.result += 1
        return self.result

在这里插入图片描述

动态规划题解做法

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def minCameraCover(self, root: Optional[TreeNode]) -> int:
        '''
            状态转移:
                0:无覆盖
                1:被覆盖
                2:放置摄像头
        '''
        self.result = 0
        # 定义递归函数
        def dfs(node):
            # 结点为空，当成已覆盖
            if not node:
                return 1
            # 后序遍历，判断孩子结点的顺序
            left = dfs(node.left)
            right = dfs(node.right)
            # 左右结点都被覆盖，则该结点不必覆盖
            if left == right == 1:
                return 0
            # 左右结点有一个未被覆盖，则安装相机
            elif left == 0 or right == 0:
                self.result += 1
                return 2
            # 其他情况均为被覆盖
            return 1
        root_status = dfs(root)
        # 如果root没有被覆盖，则安装相机
        if not root_status:
            self.result += 1
        return self.result