leetcode装饰树

ghblk

已于 2023-10-11 18:44:32 修改

阅读量64

点赞数

分类专栏： leetcode题解文章标签： leetcode 算法数据结构

于 2023-05-04 16:15:12 首次发布

本文链接：https://blog.csdn.net/ghblk/article/details/130489159

版权

leetcode题解专栏收录该内容

75 篇文章 1 订阅

订阅专栏

该问题涉及二叉树操作，需在每个非空节点与子节点间插入值为-1的节点作为灯饰。提供的代码示例展示了两种方法实现这一操作：一种是尾递归，另一种是直接递归。两种方法都遍历树结构，当遇到非空子节点时，插入新节点并继续递归扩展。

摘要由CSDN通过智能技术生成

力扣嘉年华上的 DIY 手工展位准备了一棵缩小版的二叉装饰树 root 和灯饰，你需要将灯饰逐一插入装饰树中，要求如下：

完成装饰的二叉树根结点与 root 的根结点值相同
若一个节点拥有父节点，则在该节点和他的父节点之间插入一个灯饰（即插入一个值为 -1 的节点）。具体地：
在一个父节点 x 与其左子节点 y 之间添加 -1 节点，节点 -1、节点 y 为各自父节点的左子节点，
在一个父节点 x 与其右子节点 y 之间添加 -1 节点，节点 -1、节点 y 为各自父节点的右子节点，
现给定二叉树的根节点 root ，请返回完成装饰后的树的根节点。
示例 1：

输入：
root = [7,5,6]

输出：[7,-1,-1,5,null,null,6]

解释：如下图所示，

示例 2：

输入：
root = [3,1,7,3,8,null,4]

输出：[3,-1,-1,1,null,null,7,-1,-1,null,-1,3,null,null,8,null,4]

解释：如下图所示

提示：

0 <= root.Val <= 1000
root 节点数量范围为 [1, 10^5]
链接：https://leetcode.cn/problems/KnLfVT

尾递归

class Solution {
public:
TreeNode expandBinaryTree(TreeNode* root)
{
    if(root != nullptr)
    {
        expandBinaryTree(root->left);
        if(root->left!=nullptr)
        {
            TreeNode* T = new TreeNode(-1);
            T->left = root->left;
            root->left = T;
        }
        expandBinaryTree(root->right);
        if(root->right!=nullptr)
        {
            TreeNode* T = new TreeNode(-1);
            T->right = root->right;
            root->right = T;
        }
    }
    return root;
}
};

直接递归

class Solution {
public:
TreeNode* expandBinaryTree(TreeNode* root) {

    if(root->left !=nullptr){
        TreeNode* light = new TreeNode(-1);
        light->left = root->left;
        root->left = light;
        expandBinaryTree(light->left);
    }
    if(root->right !=nullptr){
        TreeNode* light = new TreeNode(-1);
        light->right = root->right;
        root->right = light;
        expandBinaryTree(light->right);
    }
    return root;
}      
};