算法学习---监控二叉树

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

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

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

LeetCode题目链接:https://leetcode-cn.com/problems/binary-tree-cameras/

从题目中知道,我们需要求得最小摄像头数量
那么怎么样去保证 最小
我们来用一个节点的不需安装摄像头的情况列举一下

  1. 当此节点的左右子节点,一个节点为已监视状态,一个节点为已安装状态,不需安装摄像头
  2. 当此节点的左右子节点均为已安装状态,不需安装摄像头
  3. 当此节点的左右子节点均为,不需安装摄像头
  4. 当此节点的左右子节点均为已监视状态,不需安装摄像头

对于3、4情况下,不需安装摄像头的原因就是,可以交给父节点去安装,以达到监视此节点的效果
(这里其实有一些贪心思想)
但是这3、4情况有一个特例,那就是当此节点为根节点的时候,这两种情况仍然需要安装摄像头
看下图
为了便于记录
0标记为未监视
1标记为已监视
2标记为已安装
图1:
在啊啊啊啊嗄
图2:
在这里插入图片描述

我们再来将一个节点的需要安装摄像头的情况列举一下

  1. 当此节点的左右子节点其中一个为未监视,需要安装摄像头

所有情况已经列举出来,那么就显而易见了,将其转化为代码(注释中已将各个情况处理所在标出)

代码如下

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int count = 0;
    //0未监视 1已监视 2已安装
	int dfs(TreeNode* root) {
		if (root == NULL)
			return -1;
		int left = dfs(root->left), right = dfs(root->right);
		if (left == -1 && right == -1)return 0;			//不需安装情况 3
		if (left == 0 || right == 0)					//需要安装
		{
			count++;
			return 2;
		}
		if (left == 2 || right == 2)return 1;			//不需安装情况 1、2
		return 0;										//不需安装情况 4
	}
	int minCameraCover(TreeNode* root) {
		//处理根节点情况
		//dfs(root)返回0时
		//说明根节点无法被监视到
		//因此需要在根节点处加一个摄像头
		return (dfs(root) == 0) ? count + 1 : count; 
	}
};

总结:可以看到,我们在解题过程中,利用了贪心的思想,以状态变量辅助(标记已监视、未监视、已安装),最终成功解出了题目

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值