力扣刷题：dfs递归解决二叉树剪枝

最新推荐文章于 2024-07-21 08:37:44 发布

W_oilpicture

最新推荐文章于 2024-07-21 08:37:44 发布

阅读量112

点赞数 3

分类专栏：算法刷题文章标签：深度优先 leetcode 剪枝

本文链接：https://blog.csdn.net/WandZ123/article/details/125920324

版权

算法刷题专栏收录该内容

10 篇文章 0 订阅

订阅专栏

前言

人生如逆旅，我亦是行人。

在这里插入图片描述

/**
 * 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 dfs(TreeNode* root){
        if(!root) return false;         //空节点代表没有1，返回false
        bool left = dfs(root->left);    //判断左边有没有1
        bool right = dfs(root->right);  //判断右边有没有1
        if(!left)   root->left = NULL;  //左边完全没有1，则删除左子树
        if(!right)  root->right = NULL; //右边完全没有1，则删除右子树
        if(root->val != 1 && !left && !right)   
            return false;   //左边没有1而且右边也没有1并且自己也不是1，应该删除（返回false）
        return true;

    }
    TreeNode* pruneTree(TreeNode* root) {
        bool valid = dfs(root);
        if(!valid)
            root = nullptr; // 若整个树都没有1则返回空节点
        return root;
    }
};

/**
 * 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 seek(TreeNode* root)
    {
        if(root == NULL)
            return true;
        if(!seek(root->left))
            root->left = NULL;  //
        if(!seek(root->right))
            root->right = NULL;
        return root->val || root->left || root->right;//如果有孩子不为空(孩子含有1),或这个结点为1都不能删除这个结点
    }
    TreeNode* pruneTree(TreeNode* root) {
        if(seek(root))
            return root;
        return NULL;
    }
};