【LeetCode每日一题】[困难]968. 监控二叉树(相机覆盖)
968. 监控二叉树
题目来源link
算法思想:树,递归,贪心
题目简述:
给定一个二叉树,我们在树的节点上安装摄像头。
节点上的每个摄影头都可以监视其父对象、自身及其直接子对象。
计算监控树的所有节点所需的最小摄像头数量。
题目分析
分析来源link
这题正常的逻辑是从上往下,但我们还可以逆向思维,从下往上来统计。那么一个节点就会有3种情况:
1,当前节点有相机 HAS_CAM
2,当前节点不需要相机(子节点有相机把它给覆盖了)NO_NEED_CAM
3,当前节点没有相机并且也没有被子节点给覆盖(那么他只能等他的父节点把它给覆盖了)NO_CAM
java代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
private int res = 0;
private final static int NO_CAM = 0; //没有相机,需要覆盖它
private final static int HAS_CAM = 1;//我是相机
private final static int NO_NEED_CAM = 2;//没有相机,但是不需要覆盖它(已覆盖)
public int minCameraCover(TreeNode root) {
int temp = dfs(root);
if (temp == NO_CAM) {
//对该节点进行判断,是否要添加相机
res++;
}
return res;
}
private int dfs(TreeNode root) {
if (root == null) {
//如果是空节点,则不需要覆盖(相机)
return NO_NEED_CAM;
}
int l = dfs(root.left);//递归左子树
int r = dfs(root.right);//递归右子树
//左右子树只要有一个要覆盖的,这个节点就必须放相机,从而实现覆盖
if (l == NO_CAM || r == NO_CAM) {
res++;
return HAS_CAM;//该节点放入相机,状态为HAS_CAM
}
//两边只有有一个相机,该节点就不需要相机,已经覆盖
if (l == HAS_CAM || r == HAS_CAM) {
return NO_NEED_CAM;//该节点已经覆盖,不需要相机
}
//其他情况 : 即两边都是NO_NEED_CAM,都不需要覆盖,那么该节点就是没有相机覆盖的点,
//该节点需要别人覆盖(贪心),尽可能的少相机数量,所以该阶段需要其父亲节点来覆盖
return NO_CAM;//该节点,需要父亲节点来覆盖
}
}