动态规划 leetcode 968. 监控二叉树

题目：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;
    }
};