LeetCode | 0623. 在二叉树中增加一行【Python】

素颜马尾好姑娘i 2022-11-03 00:43 128阅读 0赞

## 问题 ##

[力扣][Link 1]

给定一个二叉树，根节点为第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.  输入的二叉树至少有一个节点。

## 思路 ##

**递归**

d = 1：将原先整棵树作为 v 的左子树
    d = 2：把左子树接到新的左节点的左子树上，把右子树接到新的右节点的右子树上
    d > 2：分解

## 代码 ##

### Python3 ###

# Definition for a binary tree node.
    # class TreeNode:
    # def __init__(self, x):
    # self.val = x
    # self.left = None
    # self.right = None
    
    class Solution:
        def addOneRow(self, root: TreeNode, v: int, d: int) -> TreeNode:
            if not root:
                return
            
            # 将原先整棵树作为 v 的左子树
            if d == 1:
                node = TreeNode(v)
                node.left = root
                return node
            
            if d == 2:
                # 新左节点
                L = TreeNode(v)
                # 新右节点
                R = TreeNode(v)
    
                L.left = root.left
                R.right = root.right
    
                root.left = L
                root.right = R
                return root
            
            # d > 2 不断分解
            self.addOneRow(root.left, v, d - 1)
            self.addOneRow(root.right, v, d - 1)
            return root

## 链接 ##

[GitHub][]

[Link 1]: https://leetcode-cn.com/problems/add-one-row-to-tree/
[GitHub]: https://github.com/Wonz5130/LeetCode-Solutions/tree/master/solutions/0623-Add-One-Row-to-Tree