题目描述
En:
Given a binary tree,we install cameras on the nodes of tree.
Each camera at a node can monitor its parent,itself and its immediate children.
Calculate the minimum number of cameras needed to monitor all nodes of the tree.
中文:
给定一个二叉树,我们在树的节点上安装摄像头。
节点上的每个摄影头都可以监视其父对象、自身及其直接子对象。
计算监控树的所有节点所需的最小摄像头数量。
题解
我们可以为每个节点设置三个状态:
- 0:安装了监视器
- 1:没有安装监视器但可见
- 2:不可见,则需要安装监视器
可以通过动态规划,设当前节点为root,则有两种情况:
- 在root处安装监视器,则需保证左右孩子的子数被覆盖
- 在root不安装监视器,则left和right两个孩子处则需安装监视器
可以通过后序遍历来实现:
int ans=0;
int dfs(TreeNode* node)
{
if(node==NULL)
return 1;
int left=dfs(node->left),right=dfs(node->right);
//当两个孩子之一不可见时,则安装监视器
if(left==2 || right==2){
ans++;
return 0;
}
//当两个孩子之一安装了监视器时,则设为可见状态
else if(left==0 || right==0)
{
return 1;
}else
return 2;
}
int minCameraCover(TreeNode* root) {
if(root==NULL)
return 0;
if(dfs(root)==2)
ans++;
return ans;
}