《代码随想录》Ⅷ 贪心算法 968. 监控二叉树

最新推荐文章于 2025-12-18 20:25:19 发布

原创最新推荐文章于 2025-12-18 20:25:19 发布 · 475 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#C++ #贪心算法

《代码随想录》Ⅷ 贪心算法 968. 监控二叉树

‍

革命尚未成功，同志仍需努力！

题目：力扣链接

给定一个二叉树，我们在树的节点上安装摄像头。

节点上的每个摄影头都可以监视其父对象、自身及其直接子对象。

计算监控树的所有节点所需的最小摄像头数量。

一、思想

这道题的核心是使用贪心算法+后序遍历来确定摄像头的最小安装位置。具体思路如下：

状态定义：
- 0：该节点无覆盖
- 1：该节点有摄像头
- 2：该节点有覆盖（但无摄像头）
后序遍历策略：
- 从叶子节点开始向上处理
- 优先让叶子节点的父节点安装摄像头（贪心思想）
状态转移规则：
- 左右子节点都有覆盖时，当前节点不安装（状态0）
- 任一子节点无覆盖时，当前节点必须安装（状态1）
- 任一子节点有摄像头时，当前节点获得覆盖（状态2）

二、代码

class Solution
{
public:
    int result;
    int traversal(TreeNode *root)
    {
        if (root == nullptr)
        {
            return 2; // 空节点，该节点有覆盖
        }

        int left = traversal(root->left);   // 左子树的状态
        int right = traversal(root->right); // 右子树的状态

        // 情况1：左右子树都有覆盖
        if (left == 2 && right == 2)
        {
            return 0; // 本节点无覆盖
        }
        // 情况2：左右子树至少有一个无覆盖
        else if (left == 0 || right == 0)
        {
            result++; // 本节点需要放置摄像头
            return 1; // 本节点需要放置摄像头
        }
        // 情况4：左右子树都有覆盖
        else if (left == 1 || right == 1)
        {
            return 2; // 本节点有覆盖
        }

        return -1; // 其他情况，返回-1，这是为了防止代码逻辑错误
    }

    int minCameraCover(TreeNode *root)
    {
        result = 0; // 初始化摄像头数量
        if (traversal(root) == 0)
        {
            result++; // 根节点无覆盖，需要放置摄像头
        }
        return result;
    }
};