JZ0047：二叉树剪支

title:二叉树剪支

📃 题目描述

题目链接：二叉树剪支、相同题目

🔔 解题思路

第一种做法：dfs，思考方式是你站在一个点上怎么做，在一个点上，先看左子树是否要删除，再看看右子树，然后进行删除操作，最后看看自己是不是属于要删除的节点；

class Solution {
public:
TreeNode* pruneTree(TreeNode* root) {
return deleteNode(root) ? nullptr : root;
}

bool deleteNode(TreeNode* root) {
if (root == nullptr) return false;
bool leftDel = deleteNode(root->left);
bool rightDel = deleteNode(root->right);

if (leftDel) root->left = nullptr;
if (rightDel) root->right = nullptr;

// if (root->left == nullptr && root->right == nullptr && root->val == 0) return true;
// else return false;
//优化写法：
return root->val == 0 && !root->left && !root->right;
}
};

方法二：非递归方式实现，将数据压入栈中，每次拿出，判断这个节点的左孩子、右孩子是不是要删除，需要删除的就删除；

💥 复杂度分析

• 时间复杂度：o(n);
• 空间复杂度：O(n);