814. 二叉树剪枝
这题华师数据科学学院的机试题
class Solution {
public:
TreeNode* pruneTree(TreeNode* root) {
if(!dfs(root))return NULL; // 整个树都全是0
return root;
}
bool dfs(TreeNode *root){
if(!root)return false;
bool left=dfs(root->left);
bool right=dfs(root->right);
if(!left)root->left=NULL;
if(!right)root->right=NULL;
return root->val==1 || left || right;
}
};









网友评论