leecode 装饰树

题目

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

思路

不知道为啥leecode上通过率很低，DFS一下，时间复杂度O(n) n节点数，空间复杂度O（h）h是树高度，发现非-1子节点直接插入然后继续给子节点做搜索就好了。

代码

void addMinus1Son(struct TreeNode* root) {
    if (root == NULL) return;

    // 左子树
    if (root->left && root->left->val != -1) {
        struct TreeNode* newLeft = (struct TreeNode*)malloc(sizeof(struct TreeNode));
        newLeft->val = -1;
        newLeft->left = root->left;
        newLeft->right = NULL;
        root->left = newLeft;
        addMinus1Son(newLeft->left);
    }
    
    // 右子树
    if (root->right && root->right->val != -1) {
        struct TreeNode* newRight = (struct TreeNode*)malloc(sizeof(struct TreeNode));
        newRight->val = -1;
        newRight->right = root->right;
        newRight->left = NULL;
        root->right = newRight;
        addMinus1Son(newRight->right);
    }
}

struct TreeNode* expandBinaryTree(struct TreeNode* root) {
    addMinus1Son(root);
    return root;
}

结果