题目
给定一个二叉树,我们在树的节点上安装摄像头。
节点上的每个摄影头都可以监视其父对象、自身及其直接子对象。
计算监控树的所有节点所需的最小摄像头数量。
示例 1:
输入:[0,0,null,0,0]
输出:1
解释:如图所示,一台摄像头足以监控所有节点。
示例 2:
输入:[0,0,null,0,null,0,null,null,0]
输出:2
解释:需要至少两个摄像头来监视树的所有节点。 上图显示了摄像头放置的有效位置之一。
提示:
给定树的节点数的范围是 [1, 1000]。
每个节点的值都是 0。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-cameras
思路
题目前提:1、每个节点的值都是0;
基本思路:使用普通的二叉树遍历就可以,从叶子节点开始往上计算Camera。
有两种情况需要有Camera:
1、叶子节点的母节点为Camera;
2、非叶子节点,若子节点都不为Camera,则母节点为Camera;
程序
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
int getMinCamera(struct TreeNode* node,int* num){
int ilret=0, irret=0;
/*若为叶子节点,则返回一个camera,表示需要一个Camera进行监控*/
if(node->left == NULL && node->right == NULL){
return 1;
}
/*若子节点不为空,判断字节点的Camera情况*/
if(node->left != NULL){
ilret = getMinCamera(node->left,num);
}
if(node->right != NULL){
irret = getMinCamera(node->right,num);
}
/*若左子节点右子节点的返回值之中存在一个大于0的,则该节点为一个Camera*/
if(ilret+irret>0){
(*num)++;
node->val++;
return 0;
}
/*左右子节点返回值都为0, 且左右子节点都没有Camera 则返回一个Camera */
if(((node->left != NULL && node->left->val == 0)|| node->left == NULL) &&
((node->right != NULL && node->right->val == 0)|| node->right == NULL)){
return 1;
}
return 0;
}
int minCameraCover(struct TreeNode* root){
int num=0,iret;
if(root == NULL){
return 0;
}
iret = getMinCamera(root,&num);
/*若根节点返回了一个Camera,需要加上来*/
return num + iret;
}