给定一个二叉树,我们在树的节点上安装摄像头。
节点上的每个摄影头都可以监视其父对象、自身及其直接子对象。
计算监控树的所有节点所需的最小摄像头数量。
示例 1:
输入:[0,0,null,0,0]
输出:1
解释:如图所示,一台摄像头足以监控所有节点。
示例 2:
输入:[0,0,null,0,null,0,null,null,0]
输出:2
解释:需要至少两个摄像头来监视树的所有节点。 上图显示了摄像头放置的有效位置之一。
提示:
给定树的节点数的范围是 [1, 1000]。
每个节点的值都是 0。
基本思路:其本质是贪心,首先,我们有一个朦胧的想法,不要在叶节点安装摄像头。具体展开是:我们尽量让父节点层安装摄像头,因为父节点的个数一定小于等于子节点的个数,然后从叶节点向上递推,根据左右节点的情况,确定父节点的情况,具体就是采取后序遍历。
- 我们把当前节点的状态分为三种,0:未被覆盖(需要安装);1:安装摄像头(覆盖父节点和子节点);2:被覆盖(子节点已经安装摄像头)
- 当前节点的子节点有有未被覆盖的,当前节点的状态只能是1
- 当前节点的子节点都是已经被覆盖的,而当前节点的状态是未被覆盖的,我们可以采用两种方式,一种是安装摄像头,另一种是有父节点安装摄像头,由贪心策略知,应采用第二种方式,故当前节点的状态是0;
- 其他情况,当前节点的子节点的组合情况只能是(1,2),(2,1),(1,1),此时当前节点被覆盖,不需要安装摄像头,故状态是2
- 注意根节点的状态,如果根节点的状态是0,记得要安装摄像头。
int ans=0;
int minCameraCover(TreeNode* root) {
if(dfs(root)==0)
ans++;
return ans;
}
//0:该节点无覆盖
//1:本节点有摄像头
//2:本节点有覆盖
int dfs(TreeNode *root){
if(root==nullptr){
return 2;
}
int left=dfs(root->left);
int right=dfs(root->right);
if(left==2&&right==2){
return 0;
}
if(left==0||right==0){
ans++;
return 1;
}
if(left==1||right==1){
return 2;
}
return 0;
}