代码随想录贪心算法——监控二叉树

题目

给定一个二叉树,我们在树的节点上安装摄像头。
节点上的每个摄影头都可以监视其父对象、自身及其直接子对象。
计算监控树的所有节点所需的最小摄像头数量。
示例 1:
在这里插入图片描述
输入:[0,0,null,0,0]
输出:1
解释:如图所示,一台摄像头足以监控所有节点。
示例 2:
在这里插入图片描述
输入:[0,0,null,0,null,0,null,null,0]
输出:2
解释:需要至少两个摄像头来监视树的所有节点。 上图显示了摄像头放置的有效位置之一。
提示:
给定树的节点数的范围是 [1, 1000]。
每个节点的值都是 0。

思路

首先摄像头都没有放在叶子节点上面,摄像头可以覆盖上中下三层,如果摄像头放在叶子节点上,就会浪费一层的覆盖。

这里要注意,从头结点看起的话,头结点放不放摄像头也就省下一个摄像头,叶子节点放不放摄像头省下的摄像头是指数级别的

所以本题要从下往上看,
局部最优: 让叶子结点的父节点安摄像头,所用摄像头最少
全局最优: 全部摄像头数量所用最少

算法核心:从底到上,先给叶子节点的父节点放个摄像头,然后隔两个节点放一个摄像头,直到二叉树头结点

这里要注意两个难点:

  1. 二叉树的遍历
  2. 如何隔两个节点放一个摄像头

确定遍历顺序:后序遍历

int traversal(TreeNode cur){
	if(终止条件) return;
	
	int left = traversal(cur.left);	  //左
	int right = traversal(cur.right); //右
	
	逻辑处理							  //中
	return;
}

隔两个节点放一个摄像头:
这里需要状态转移的公式(并不是DP的状态转移公式),本题状态转移没有择优的过程,就是单纯的状态转移

每个节点可能有几种状态:

  • 该节点无覆盖
  • 该节点有摄像头
  • 该节点有覆盖

分别用三个数字来表示:

  • 0:该节点无覆盖
  • 1:该节点有摄像头
  • 2:该节点有覆盖

用通俗的话来说就是:
无论是空节点还是其他节点,都向上传递

  • 0:没人管(无覆盖)
  • 1:不用管(有摄像头)
  • 2:有人管(有覆盖)

注意:叶子结点一定是被覆盖的,不能在叶子结点安装摄像头

在遍历的时候遇到空节点时,需要判断空节点的状态,既不能是无覆盖的状态(这样叶结点就要放摄像头了),空节点也不能是有摄像头的状态(这样叶节点的父节点就没有必要放摄像头了),所以空节点只能是有覆盖的情况(即叶节点的父节点放摄像头)

递归的终止条件应该是遇到了空节点,并返回2(有覆盖/有人管)

//空节点,该节点有覆盖
if(cur == null) return 2;

单层逻辑处理:
主要有四种情况:

1. 左右节点都有覆盖
左孩子有覆盖(有人管),右孩子有覆盖(有人管),那么中间节点应该是无覆盖(没人管),因为遍历是从下往上(品,你细品,品不出来就画图

if(left == 2 && right== 2) return 0;

2. 左右节点至少有一个无覆盖的情况
这种情况中间节点(父节点)就需要放摄像头,因为毕竟有一个孩子没有覆盖到

left == 0 && right == 0//左右节点无覆盖
left == 1 && right == 0//左节点有摄像头,右节点无覆盖
left == 0 && right == 1//左节点有无覆盖,右节点摄像头
left == 0 && right == 2 //左节点无覆盖,右节点覆盖
left == 2 && right == 0//左节点覆盖,右节点无覆盖

//整合一下表示为:
if (left == 0 || right == 0) {
    result++;//摄像头个数要加一
    return 1;//表示中间节点放摄像头
}

3. 左右节点至少有一个有摄像头

如果是以下情况,其实就是左右孩子节点有一个有摄像头了,那么其父节点就应该是2(覆盖的状态)

left == 1 && right == 2//左节点有摄像头,右节点有覆盖 
left == 2 && right == 1//左节点有覆盖,右节点有摄像头 
left == 1 && right == 1//左右节点都有摄像头

//整合如下:
if (left == 1 || right == 1) return 2;

4. 头结点没有覆盖
递归结束之后,可能头结点 还有一个无覆盖的情况

    if (traversal(root) == 0) { // root 无覆盖
        result++;
    }

java代码如下:

class Solution {
	int res = 0;
	public int minCameraCover(TreeNode root){
		if(postorder(root) == 0){// 对根节点的状态做检验,防止根节点是无覆盖状态 
			res++;
		}
		return res;
	}
		
	public int postorder(TreeNode root){
		if(root == null){// 空节点默认为有覆盖状态,避免在叶子节点上放摄像头
			return 2;
		}
		int left = postorder(root.left);
		int right = postorder(root.right);
	
		if(left == 2 && right == 2){
			return 0;
		} else if(left == 0 || right == 0){
			res++;
			return 1;
		}else {
			return 2;
		}
	}
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

HDU-五七小卡

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值