leetcode_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]。
• 输入的二叉树至少有一个节点。

核心代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def addOneRow(self, root: TreeNode, val: int, depth: int) -> TreeNode:
        que = [root]
        level = 1
        d = depth
        v = val
        if d == 1:
            root_new = TreeNode(v)
            root_new.left = root
            return root_new
        while que:
            if level == d - 1:
                for i in range(len(que)):
                    inputs_node_left = TreeNode(v)
                    inputs_node_right = TreeNode(v)
                    node = que.pop(0)
                    temp_left = None
                    temp_right = None
                    if node.left:
                        temp_left = node.left
                        node.left = inputs_node_left
                        inputs_node_left.left = temp_left
                    else:
                        node.left = inputs_node_left
                    if node.right:
                        temp_right = node.right
                        node.right = inputs_node_right
                        inputs_node_right.right = temp_right
                    else:
                        node.right = inputs_node_right
            else:
                level += 1
                for i in range(len(que)):
                    node = que.pop(0)
                    if node.left:
                        que.append(node.left)
                    if node.right:
                        que.append(node.right)
        return root

解题思路：第一个部分：如果 d 的值为 1，深度 d - 1 不存在，则创建一个新的根节点 v，原先的整棵树将作为 v 的左子树，第二个部分：将 N 原先的左子树，连接为新节点 v 的左子树；将 N 原先的右子树，连接为新节点 v 的右子树，我们在d - 1的这一层就要完成增加下一层的任务，有点像在链表中添加节点一样，先断开下节点，在中间连上当前节点，在当前节点上挂载上下节点，完成增加的任务。