python二叉树增加节点_LeetCode | 0623. 在二叉树中增加一行【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

注意:

输入的深度值 d 的范围是：[1，二叉树最大深度 + 1]。

输入的二叉树至少有一个节点。

思路

递归

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

链接