LeetCode 623 在二叉树中增加一行 Python

给定一个二叉树，根节点为第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

注意:

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

   class Solution:
       def addOneRow(self, root, v, d):
           """
           :type root: TreeNode
           :type v: int
           :type d: int
           :rtype: TreeNode
           """
           if d == 1:
               tmp = TreeNode(v)
               tmp.left = root
               return tmp
           
           #讨论d>1的情形
           queue = [(root,1)]
           #首先要将所有d-1层的节点都加进来
           while queue:
               item = queue.pop(0)
               (TmpNode,depth) = item
               if depth >= d-1: 
                   queue.insert(0,(TmpNode,depth))
                   break
               if TmpNode.left: queue.append((TmpNode.left,depth+1))
               if TmpNode.right: queue.append((TmpNode.right,depth+1))
           #此时所有depth=d-1的节点都放进了queue
           
           for (node,depth) in queue:
               tmp = node.left
               mynode = TreeNode(v)
               node.left = mynode
               mynode.left = tmp
               
               tmp = node.right
               mynode = TreeNode(v)
               node.right = mynode
               mynode.right = tmp
           
           return root