【树形动态规划】968. 监控二叉树——思路分析
968. 监控二叉树 给定一个二叉树,我们在树的节点上安装摄像头。 节点上的每个摄影头都可以监视其父对象、自身及其直接子对象。 计算监控树的所有节点所需的最小摄像头数量。
输入:[0,0,null,0,0] 输出:1 解释:如图所示,一台摄像头足以监控所有节点。
思路分析【树形动态规划】
本题以二叉树为背景,不难想到用递归的方式求解。本题的难度在于如何从左、右子树的状态,推导出父节点的状态。
为了表述方便,我们约定:如果某棵树的所有节点都被监控,则称该树被「覆盖」。
对于每个节点分成三种状态: 0状态:代表该节点无覆盖的情况 1状态:代表该节点放置一个摄像头的情况 2状态:代表该节点不放置摄像头,但被覆盖的情况
那么我们很明显需要对树进行后序遍历,在遍历的同时记录放置的摄像头的数量。
1、当该节点的左孩子或右孩子处于0状态,则该节点一定要放置摄像头 2、当该节点的左右孩子都处于2状态,为了尽可能少使用摄像头,需要将摄像头尽量放置在父节点上 3、当该节点的左孩子或右孩子处于1状态,则返回2;
代码如下
class Solution {
int count = 0;
public int minCameraCover(TreeNode root) {
if(dfs(root)==0){
count++;
}
return count;
}
public int dfs(TreeNode node){
if(node==null){
return 2;
}
int left = dfs(node.left);
int right = dfs(node.right);
if(left==2&&right==2){
return 0;
}
if(left==0||right==0){
count++;
return 1;
}
if(left==1||right==1){
return 2;
}
return -1;
}
}
|