二叉树的题目解题的时候无非就是明确当前节点是需要完成什么工作,然后实现当前节点的工作后,然后递归调用对应的左右子树去完成相同的工作。
写树相关的算法,简单说就是,先搞清楚当前root节点该做什么,然后根据函数定义递归调用子节点,递归调用会让孩子节点做相同的事情。
解题的时候无论是直接操作对应的二叉树得到结果还是去构建一棵新的二叉树,都是首先明确当前的root节点是需要做什么工作,然后再次去递归调用完成对应的子树的结果的处理,最终返回对应的结果子树。
class Solution {
public:
TreeNode* removeLeafNodes(TreeNode* root, int target) {
if (!root) {
return nullptr;
}
root->left = removeLeafNodes(root->left, target);
root->right = removeLeafNodes(root->right, target);
if (!root->left && !root->right && root->val == target) {
return nullptr;
}
return root;
}
};
在二叉树剪枝的时候,只需要对应的节点直接返回为空的,不需要去初始化什么delete的函数,只需要对应的root指针对于其进行操作,不需要完成对应的节点的处理。所以不需要对应的节点操作的!