·题目描述
给定一个二叉树,我们在树的节点上安装摄像头。
节点上的每个摄影头都可以监视其父对象、自身及其直接子对象。
计算监控树的所有节点所需的最小摄像头数量。
示例 1:
输入:[0,0,null,0,0] 输出:1 解释:如图所示,一台摄像头足以监控所有节点。
示例 2:
输入:[0,0,null,0,null,0,null,null,0] 输出:2 解释:需要至少两个摄像头来监视树的所有节点。 上图显示了摄像头放置的有效位置之一。
提示:
- 给定树的节点数的范围是
[1, 1000]
。 - 每个节点的值都是 0。
·解题思路
看的代码随想录的解题思路,太牛了!!
1从叶子节点往上遍历(后续遍历),根据叶子节点的状态确定父节点的状态.在中的时候处理逻辑
2. 0 ——表示该结点无覆盖;1——表示该节点有覆盖; 2——表示该节点安装摄像头;
3.根据返回值确定父节点状态
1)左孩子或者右孩子 有摄像头的时候,父节点被覆盖,状态为1
2)左孩子和右孩子均被覆盖的时候,父节点不需要安装摄像头,状态为1 无覆盖
3)左孩子或者右孩子无覆盖的时候,父节点需要安装摄像头,状态为2
4. 空结点不需要做特殊处理,表示为有覆盖即可
5.还有一种特殊的情况,就是当父节点的状态为有覆盖,并且到达根节点的时候,需要在根结点安装一个摄像头覆盖该节点。
·Java代码
class Solution {
public int count = 0;
public int minCameraCover(TreeNode root) {
int res = dfs(root);
if(res == 0) count ++;//根节点没有覆盖,等待更高级覆盖,所以需要安装摄像头
return count;
}
public int dfs(TreeNode root){
if(root == null) return 1; //当结点为空结点的时候不需要做过多的处理,只需要表示为有覆盖的状态即可
int left = dfs(root.left); //左孩子遍历
int right = dfs(root.right);//遍历右孩子
//中的处理逻辑
if(left == 0 || right == 0) {
count ++;
return 2; //左或者右无覆盖,父节点安装摄像头覆盖子节点
}
if(left == 1 && right == 1) return 0;//左右孩子都被覆盖,父节点表示为0,等待更高一个父节点覆盖
if(left == 2 || right == 2) return 1;//左孩子或者右孩子右摄像头,父节点被覆盖
return -1;
}
}