【树-中等】623. 在二叉树中增加一行

【题目】
给定一个二叉树，根节点为第1层，深度为 1。在其第 d 层追加一行值为 v 的节点。

添加规则：给定一个深度值 d （正整数），针对深度为 d-1 层的每一非空节点 N，为 N 创建两个值为 v 的左子树和右子树。

将 N 原先的左子树，连接为新节点 v 的左子树；将 N 原先的右子树，连接为新节点 v 的右子树。

如果 d 的值为 1，深度 d - 1 不存在，则创建一个新的根节点 v，原先的整棵树将作为 v 的左子树。

示例 1:

输入:
二叉树如下所示:
4
/
2 6
/ \ /
3 1 5

v = 1

d = 2

输出:
4
/
1 1
/
2 6
/ \ /
3 1 5

示例 2:

输入:
二叉树如下所示:
4
/
2
/ \
3 1

v = 1

d = 3

输出:
4
/
2
/ \
1 1
/ \
3 1

注意:

输入的深度值 d 的范围是：[1，二叉树最大深度 + 1]。
输入的二叉树至少有一个节点。

【代码】

class Solution:
    def dfs(self,root,dep):
        if not root:
            return 
        if dep==self.depth:
            self.level.append(root)
        self.dfs(root.left,dep+1)
        self.dfs(root.right,dep+1)
    def addOneRow(self, root: TreeNode, val: int, depth: int) -> TreeNode:
        if depth==1:
            newHead=TreeNode(val)
            newHead.left=root
            return newHead
        self.level=[]
        self.depth=depth-1
        self.dfs(root,1)
        for node in self.level:
            nodeLeft=TreeNode(val)
            nodeRight=TreeNode(val)
            nodeLeft.left=node.left
            nodeRight.right=node.right
            node.left=nodeLeft
            node.right=nodeRight
        return root

【方法2】

class Solution:
    def dfs(self,root,dep):
        if not root:
            return 
        if dep==self.depth:
            node=root
            nodeLeft=TreeNode(self.val)
            nodeRight=TreeNode(self.val)
            nodeLeft.left=node.left
            nodeRight.right=node.right
            node.left=nodeLeft
            node.right=nodeRight
        else:
            self.dfs(root.left,dep+1)
            self.dfs(root.right,dep+1)
    def addOneRow(self, root: TreeNode, val: int, depth: int) -> TreeNode:
        if depth==1:
            newHead=TreeNode(val)
            newHead.left=root
            return newHead
        self.val=val
        self.depth=depth-1
        self.dfs(root,1)
        return root