最近刷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
注意:
- 输入的深度值 d 的范围是:[1,二叉树最大深度 + 1]。
- 输入的二叉树至少有一个节点。
--------------------------------------------------------------------------------------
这个题目中等难度,先对思路还是比较简单,就是从根结点开始遍历,然后深度值符合要求的地方插入对应的书结点,具体实现如下
扫描二维码关注公众号,回复:
2085574 查看本文章
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); } } }