【LCP 67. 装饰树】Leetcode C++实现

题目描述

力扣嘉年华上的 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]

---

代码实现：

【以递归的的方式走一遍树，如果当前的树R有左子树A或右子树B，才需要插入-1，子树A和B需要把自身先弄利索了，返回给R，所以递归地往下一直深搜。当左右子树不存在时（例如叶子节点），R不会被修改，直接就返回了】
【通过这个题学习一下树节点的三种构造函数形式】

/**
 * 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:
    TreeNode* expandBinaryTree(TreeNode* root) {

        if(root==nullptr) return nullptr;
        
        if(root->left!=nullptr){ 
            TreeNode* left=expandBinaryTree(root->left); //先把你的左子树都弄好了
            TreeNode* addNode=new TreeNode(-1,left,nullptr); //我来帮你在脑袋上插个-1
            root->left=addNode;
        }
        if(root->right!=nullptr){
            TreeNode* right=expandBinaryTree(root->right); //先把你的右子树都弄好了
            TreeNode* addNode=new TreeNode(-1,nullptr,right); //我来帮你在脑袋上插个-1
            root->right=addNode;
        }
        return root; 
    }
};