【leetcode】623. Add One Row to Tree

最新推荐文章于 2024-10-28 20:59:44 发布

啾啾_yeah

最新推荐文章于 2024-10-28 20:59:44 发布

阅读量204

点赞数

文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/qq_36294507/article/details/113705196

版权

该博客主要介绍了如何使用Python实现LeetCode上的一个问题——在给定深度上向二叉树添加一行节点。解决方案基于双端队列和广度优先搜索，通过层序遍历在指定深度插入新节点。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

https://leetcode.com/problems/add-one-row-to-tree/

题目大意：

在某一层插入一行节点

解题思路：

基于双端队列、广度优先搜索的二叉树层序遍历

python代码：

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def addOneRow(self, root, v, d):
        """
        :type root: TreeNode
        :type v: int
        :type d: int
        :rtype: TreeNode
        """
        if d==1:
            new_root = TreeNode(v)
            new_root.left = root
            return new_root
        
        Q = deque()
        Q.append((root,1))
        while Q:
            node,level = Q.popleft()
            if level == d-1:
                temp_left = node.left
                temp_right = node.right
                node.left = TreeNode(v)
                node.left.left = temp_left
                node.right = TreeNode(v)
                node.right.right = temp_right
            else:
                if node.left:
                    Q.append((node.left,level+1))
                if node.right:
                    Q.append((node.right,level+1))
        
        return root