# [LeetCode 814. Binary Tree Pruning ](https://leetcode.com/explore/challenge/card/july-leetcoding-challenge-2021/611/week-4-july-22nd-july-28th/3824/) ### Daily challenge Jul 23, 2021 (MEDIAN) >Given the `root` of a binary tree, return the same tree where every subtree (of the given tree) not containing a `1` has been removed. > >A subtree of a node node is node plus every node that is a descendant of node. ![](https://i.imgur.com/6sTCrD3.png) :::info **Example 1:** **Input:** root = [1,null,0,0,1] **Output:** [1,null,0,null,1] **Explanation:** Only the red nodes satisfy the property "every subtree not containing a 1". The diagram on the right represents the answer. ::: ![](https://i.imgur.com/yQap3ux.png) :::info **Example 2:** **Input:** root = [1,0,1,0,0,0,1] **Output:** [1,null,1,null,1] ::: ![](https://i.imgur.com/tvAC86s.png) :::info **Example 3:** **Input:** root = [1,1,0,1,1,0,1,0] **Output:** [1,1,0,1,1,null,1] ::: :::warning **Constraints:** * The number of nodes in the tree is in the range [1, 200]. * Node.val is either 0 or 1. ::: --- ### Approach 1 : Recursion :bulb: **`4 ms ( % )`** **`O()`** 首先走到 `tree` 的最底層,再慢慢向上層判斷。 >1. 若 **`root == NULL`** ---> 回傳 `NULL`。 >2. 若 **`root->val == 0 && root->left == NULL && root->right == NULL`**。 >---> 表示該 `root` 應該被 `remove`,回傳 `NULL`。 >3. Otherwise ---> 回傳 `root`。 ```cpp=1 /** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode() : val(0), left(nullptr), right(nullptr) {} * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {} * }; */ class Solution { public: TreeNode* cut(TreeNode* root) { if(root == NULL) return NULL; root->left = cut(root->left); root->right = cut(root->right); if(root->val == 0 && root->left == NULL && root->right == NULL) return NULL; return root; } TreeNode* pruneTree(TreeNode* root) { return cut(root); } }; ```