leetcode 968. 监控二叉树

leetcode 968. 监控二叉树

题目详情

题目链接
给定一个二叉树，我们在树的节点上安装摄像头。
节点上的每个摄影头都可以监视其父对象、自身及其直接子对象。
计算监控树的所有节点所需的最小摄像头数量。

• 示例 1：
  
  输入：[0,0,null,0,0]
  输出：1
  解释：如图所示，一台摄像头足以监控所有节点。
• 示例 2：
  
  输入：[0,0,null,0,null,0,null,null,0]
  输出：2
  解释：需要至少两个摄像头来监视树的所有节点。 上图显示了摄像头放置的有效位置之一。
  提示：
  给定树的节点数的范围是 [1, 1000]。
  每个节点的值都是 0。

我的代码

// 不会做。。。

执行用时为 40 ms 的范例

/**
 * 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:
    int minCameraCover(TreeNode* root) {
        ans = 0;
        if (root == NULL) return 0;
        if (dfs(root) == 0) ans++;
        return ans;
    }
    //0 不可观测 // 1 可被观测，未装摄像头 2 已安装
    //00, 01, 02, 10, 11, 12, 20, 21, 22
    int dfs(TreeNode* root) {
        if (root == NULL) return 1;
        int left = dfs(root->left), right = dfs(root->right);
        if (left == 0 || right == 0) {
            ans++;
            return 2;
        } else if(left + right >= 3){
            return 1;
        }else {
            return 0;
        }
    }
private:
    int ans = 0;
};

思考

参见官方解答