0
点赞
收藏
分享

微信扫一扫

第 8 天_广度优先搜索 / 深度优先搜索【算法入门】


第 8 天_广度优先搜索 / 深度优先搜索【算法入门】

  • ​​617. 合并二叉树​​
  • ​​题解​​
  • ​​116. 填充每个节点的下一个右侧节点指针​​
  • ​​官方​​

617. 合并二叉树

给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。

你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。

示例 1:

输入: 
Tree 1 Tree 2
1 2
/ \ / \
3 2 1 3
/ \ \
5 4 7
输出:
合并后的树:
3
/ \
4 5
/ \ \
5 4 7

注意: 合并必须从两个树的根节点开始。

题解

思路
递归

/**
* 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 mergeTrees(TreeNode root1, TreeNode root2) {

if(root2==null){
return root1;
}
if(root1==null){
return root2;
}
TreeNode merged = new TreeNode(root1.val + root2.val);
merged.left=mergeTrees(root1.left,root2.left);
merged.right=mergeTrees(root1.right,root2.right);
return merged;
}
}

第 8 天_广度优先搜索 / 深度优先搜索【算法入门】_递归

116. 填充每个节点的下一个右侧节点指针

给定一个 完美二叉树 ,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}

填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。

初始状态下,所有 next 指针都被设置为 NULL。

进阶:

  • 你只能使用常量级额外空间。
  • 使用递归解题也符合要求,本题中递归程序占用的栈空间不算做额外的空间复杂度。

示例:

第 8 天_广度优先搜索 / 深度优先搜索【算法入门】_算法_02

输入:root = [1,2,3,4,5,6,7]
输出:[1,#,2,3,#,4,5,6,7,#]
解释:给定二叉树如图 A 所示,你的函数应该填充它的每个 next 指针,以指向其下一个右侧节点,如图 B 所示。序列化的输出按层序遍历排列,同一层节点由 next 指针连接,'#' 标志着每一层的结束。

提示:

  • 树中节点的数量少于 4096
  • -1000 <= node.val <= 1000

官方

class Solution {
public Node connect(Node root) {
if (root == null) {
return root;
}

// 初始化队列同时将第一层节点加入队列中,即根节点
Queue<Node> queue = new LinkedList<Node>();
queue.add(root);

// 外层的 while 循环迭代的是层数
while (!queue.isEmpty()) {

// 记录当前队列大小
int size = queue.size();

// 遍历这一层的所有节点
for (int i = 0; i < size; i++) {

// 从队首取出元素
Node node = queue.poll();

// 连接
if (i < size - 1) {
node.next = queue.peek();
}

// 拓展下一层节点
if (node.left != null) {
queue.add(node.left);
}
if (node.right != null) {
queue.add(node.right);
}
}
}

// 返回根节点
return root;
}
}

第 8 天_广度优先搜索 / 深度优先搜索【算法入门】_深度优先_03


举报

相关推荐

0 条评论