LeetCode LCP 67. 装饰树

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

解释：如下图所示，

题意是要在两个节点中间插入一个值为-1的节点，递归插入即可：

/**
 * 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) {
            return root;
        }

        expandBinaryTree(root->left);
        expandBinaryTree(root->right);
        
        if (root->left) {
            TreeNode *midNode = new TreeNode(-1, root->left, nullptr);
            root->left = midNode;
        }

        if (root->right) {
            TreeNode *midNode = new TreeNode(-1, nullptr, root->right);
            root->right = midNode;
        }

        return root;
    }
};

如果树有n个节点，此算法时间复杂度为O（n），对每个节点和其父节点之间都插入一个值为-1的节点，空间复杂度为O（lgn），主要是栈开销，栈的深度为树的高度。