-
题目:
一棵树所有节点值为0;
选择适当结点装摄像头,摄像头可以监控本身结点以及上下父子两层;
返回用最少的摄像头监控整棵树; -
思路:
由于一个结点装摄像头可以覆盖三层,因此要想用最少的摄像头覆盖,就要尽可能每个摄像头都覆盖三层,尽量不要浪费;
对于树,无非就是前中后层序遍历;
由于树类似一个三角形,选择从下往上遍历,优先在下层装,可以最大程度覆盖结点,从下往上对应后序遍历;
每个节点状态只有三种,分别用0,1,2表示:0:未覆盖 1:装摄像头 2:没装摄像头但被覆盖
用递归函数的返回值表示该节点的状态;
后序遍历:先判断终止条件(cur为叶子的孩子);cur不为空,就先递归得到左右孩子的状态,用于判断cur的状态;
1.贪心:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
private:
int cnt = 0;
//后序遍历,实现从底往顶遍历的效果
//每个节点状态只有三种,分别用0,1,2表示;
//0:未覆盖 1:装摄像头 2:没装摄像头但被覆盖
int travel(TreeNode* cur) {
if (!cur) return 2;//对于叶子节点的孩子,应该返回2;若返回0,则叶子节点就需要摄像头;若返回1,就变成了叶子节点的爷爷装摄像头而不是叶子的父亲
int l = travel(cur->left);//左
int r = travel(cur->right);//右
//中
if (l == 0 || r == 0) {//0 0; 0 1; 1 0; 0 2; 2 0;
++cnt;
return 1;
}
else if (l == 1 || r == 1) return 2; //1 1 2; 1 2 2; 2 1 2;
else return 0; //2 2 0;
}
public:
int minCameraCover(TreeNode* root) {
int rootVal = travel(root);
if (rootVal == 0) ++cnt; //根可能还没被覆盖,需要手动判断
return cnt;
}
};
2.DP:还不会,日后二刷补
- 总结:
1.选择后序的原因:从下往上装,这样最多根节点处没被覆盖到而多装一个罢了;如果从上往下装,就可能导致叶子结点没被覆盖到,而多装好几个;
2.节点状态之所以选择0,1,2:本身树的所有节点值都是0,本身可以理解为未覆盖状态;再分别用1,2代表装摄像头头和不装摄像头但被覆盖的情况;
3.因为根据单层逻辑,++cnt的动作是由没被覆盖的结点的父亲完成的,所以若根节点没被覆盖,就需要手动++cnt;