aboutsummaryrefslogtreecommitdiff
path: root/binary_tree_pruning.cpp
blob: 3bac07de1e28974a43c13da070bb1331e1df10aa (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#include "leetcode.h"

class Solution {
public:
  TreeNode *pruneTree(TreeNode *root) {
    if (root) {
      root->left = pruneTree(root->left);
      root->right = pruneTree(root->right);
      if (root->left || root->right || root->val != 0)
        return root;
    }
    delete root;
    return nullptr;
  }
};