leetcode 81four. Binary Tree Pruning 二叉树剪枝

Problem: 814. Binary Tree Pruning 二叉树剪枝

解题过程

递归，返回子树是否包含1，若左右子树都不包含1而且值不等于1，那么就可以剪枝了，需要传递parent节点的，以及判断左右，特殊情况整个树都没有1, parent=nullptr，这时候需要返回false

Code

/** * 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: bool containA(TreeNode* root, TreeNode* parent, int leftRight) { if(root == nullptr) return false; bool l = containA(root->left, root, 0); bool r = containA(root->right, root, 1); if(l==false && r==false && root->val != 1) { if(parent==nullptr) { return false; } if(leftRight == 0) parent->left = nullptr; else parent->right = nullptr; return false; } return true; } TreeNode* pruneTree(TreeNode* root) { if(root==nullptr) return nullptr; bool ret = containA(root, nullptr, -1); if(ret == false) return nullptr; return root; } };