【LeetCode刷题】NO.43---第116题

126 阅读1分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

一.题目:

116. 填充每个节点的下一个右侧节点指针 给定一个 完美二叉树 ,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:

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

填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。 初始状态下,所有 next 指针都被设置为 NULL示例 1:

image.png

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

示例 2:

输入: root = []
输出: []

提示:

  • 树中节点的数量在 [0, 212 - 1] 范围内
  • -1000 <= node.val <= 1000

二、思路分析:

这道二叉树的题目我们可以用递归的思路来解决这道题。但是我们做递归题的时候我们通常不知道怎么塑造递归的情形,直到看到了有大神给出了递归的思路:要明确函数的定义是什么,然后利用定义推导最后的结果,绝不要跳入递归的细节中

二叉树的问题就是要我们细化每个节点应该干什么事,所以对于这道题目,我们明确了把一层的每个节点填充它的next指针,每一层的最后一个节点都是null,所以我们想到将所有节点的左节点的next置为右节点,但是不同父母节点的左右节点怎么连接起来?

所以我们就需要另外构建一个辅助函数,该函数的作用是可以将第一个节点的右节点与第二个节点的左节点连接起来,在加上另外的每个子树自身的左右节点相连即可完成作答。

三、代码:

/**
 * @param {Node} root
 * @return {Node}
 */
var connect = function(root) {
    if(root == null)return null
    connectTwoNode(root.left,root.right)
    return root
};
var connectTwoNode = function(Node1,Node2){
    if(Node1 == null || Node2 == null) return
    Node1.next = Node2
    connectTwoNode(Node1.left,Node1.right)
    connectTwoNode(Node2.left,Node2.right)
    connectTwoNode(Node1.right,Node2.left)
}

四、总结:

这道二叉树的题目很经典,主要做法:递归。我们需要明确二叉树的问题我们主要就是明确每个节点需要干什么事,如果明确了这个问题那么我们做二叉树的题目就简单了很多。