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

本文提供了一种解决方案来实现在指定深度为d的位置处为二叉树添加一行节点值为v的功能。该方法通过递归遍历二叉树,并在适当位置插入新节点。

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

最近刷LeetCode题目的一些思路,题目信息

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

--------------------------------------------------------------------------------------

这个题目中等难度,先对思路还是比较简单,就是从根结点开始遍历,然后深度值符合要求的地方插入对应的书结点,具体实现如下

public TreeNode addOneRow(TreeNode root, int v, int d) {
    if(d == 1){
        TreeNode treeNode = new TreeNode(v);
        treeNode.left = root;
        return treeNode;
    }
    TreeNode retTreeNode = new TreeNode(root.val);
    setTreeNode(retTreeNode,root,2,d,v);
    return retTreeNode;
}

private void setTreeNode(TreeNode root,TreeNode treeNode,int level, int targetDeep, int value){
    if(level == targetDeep && treeNode != null){
        TreeNode leftTreeNode = new TreeNode(value);
        TreeNode rightTreeNode = new TreeNode(value);
        leftTreeNode.left = treeNode.left;
        rightTreeNode.right = treeNode.right;
        root.left = leftTreeNode;
        root.right = rightTreeNode;
    } else {
        if(treeNode != null){
            root.left = treeNode.left;
            root.right = treeNode.right;
            setTreeNode(treeNode.left,treeNode.left,level + 1,targetDeep,value);
            setTreeNode(treeNode.right,treeNode.right,level + 1,targetDeep,value);
        }
    }

}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值