题目
给定一个二叉树,我们在树的节点上安装摄像头。
节点上的每个摄影头都可以监视其父对象、自身及其直接子对象。
计算监控树的所有节点所需的最小摄像头数量。
示例 1:
输入:[0,0,null,0,0]
输出:1
解释:如图所示,一台摄像头足以监控所有节点。
示例 2:
输入:[0,0,null,0,null,0,null,null,0]
输出:2
解释:需要至少两个摄像头来监视树的所有节点。 上图显示了摄像头放置的有效位置之一。
提示:
给定树的节点数的范围是 [1, 1000]。
每个节点的值都是 0。
思路
首先摄像头都没有放在叶子节点上面,摄像头可以覆盖上中下三层,如果摄像头放在叶子节点上,就会浪费一层的覆盖。
这里要注意,从头结点看起的话,头结点放不放摄像头也就省下一个摄像头,叶子节点放不放摄像头省下的摄像头是指数级别的
所以本题要从下往上看,
局部最优: 让叶子结点的父节点安摄像头,所用摄像头最少
全局最优: 全部摄像头数量所用最少
算法核心:从底到上,先给叶子节点的父节点放个摄像头,然后隔两个节点放一个摄像头,直到二叉树头结点
这里要注意两个难点:
- 二叉树的遍历
- 如何隔两个节点放一个摄像头
确定遍历顺序:后序遍历
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;
}
}
}