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;
}
};
|