题目:https://leetcode-cn.com/problems/binary-tree-cameras/
/**
* 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 {
public:
// 从下往上确定状态,假定当前节点的子节点都被覆盖了,定义当前节点的状态
// 当前节点状态 0.没有被覆盖; 被覆盖(1.no cam, 2.have cam)
// 三种状态:
// [状态 0]:当前节点没有被覆盖,它的所有子节点都被覆盖。
// [状态 1]:当前节点和它的所有子节点都被覆盖,但是该节点没有摄像头。 no camera
// [状态 2]:当前节点和它的所有子节点都被覆盖,该节点有摄像头。 have camera
int minCameraCover(TreeNode* root) {
vector<int> res = solve(root);
return min(res[1], res[2]);
}
private:
vector<int> solve(TreeNode* root) {
if(root == NULL) {
vector<int> res = {0, 0, 9999};
return res;
}
vector<int> L = solve(root->left);
vector<int> R = solve(root->right);
int covered_L = min(L[1], L[2]);
int covered_R = min(R[1], R[2]);
// 当前node没有被覆盖,说明左右子节点处于情况1,被覆盖 但是没有camera
int status_0 = L[1] + R[1];
// 当前node被覆盖但是没有camera,则左节点或者右节点必定有cam
int status_1 = min(L[2] + covered_R, R[2] + covered_L);
// 当前node被覆盖并且有cam, 左右子树都在 status 0/1中取最优, min(覆盖,不覆盖)
int status_2 = 1 + min(L[0], covered_L) + min(R[0], covered_R);
vector<int> result = {status_0, status_1, status_2};
return result;
}
};