力扣嘉年华上的 DIY 手工展位准备了一棵缩小版的 二叉 装饰树
root
和灯饰,你需要将灯饰逐一插入装饰树中,要求如下:
- 完成装饰的二叉树根结点与
root
的根结点值相同- 若一个节点拥有父节点,则在该节点和他的父节点之间插入一个灯饰(即插入一个值为
-1
的节点)。具体地:
- 在一个 父节点 x 与其左子节点 y 之间添加 -1 节点, 节点 -1、节点 y 为各自父节点的左子节点,
- 在一个 父节点 x 与其右子节点 y 之间添加 -1 节点, 节点 -1、节点 y 为各自父节点的右子节点
现给定二叉树的根节点
root
,请返回完成装饰后的树的根节点。 示例 1:
输入:
root = [7,5,6]
输出:
[7,-1,-1,5,null,null,6]
解释:如下图所示
思路:
运用深度优先搜索BFS,即层序遍历
原理是递归遍历
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public TreeNode expandBinaryTree(TreeNode root) {
ArrayList<Integer> Nodelists=new ArrayList<>();//队列存放节点
return bfsTree(root);
}
public TreeNode bfsTree(TreeNode root){
if(root==null){
return null;
}
if(root.left!=null){
TreeNode a=root.left;
root.left=new TreeNode(-1);
root.left.left=a;
root.left.right=null;
bfsTree(a);
}
if(root.right!=null){
TreeNode a=root.right;
root.right=new TreeNode(-1);
root.right.right=a;
root.right.left=null;
bfsTree(a);
}
return root;
}
}