LeetCode 算法入门 — 填充每个节点的下一个右侧节点指针

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

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

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

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

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

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

示例 1：

输入：root = [1,2,3,4,5,6,7]
输出：[1,#,2,3,#,4,5,6,7,#]
解释：序列化的输出按层序遍历排列，同一层节点由 next 指针连接，'#' 标志着每一层的结束。

示例 2:

输入：root = []
输出：[]

思路分析

1. 边界值判定。
2. 如上图所示，若左节点 left 不为空，那么可以知道 root.left.next = root.right。那么，root.right.next 要怎么计算呢？由上图可得知，如果 root.next 存在，则为 root.next.left；若不存在，那么则为 null，即 root.right.next = root.next ? root.next.left : null。
3. 递归执行树的左节点 root.left 以及右节点 root.right 即可。

AC 代码

/**
 * // Definition for a Node.
 * function Node(val, left, right, next) {
 *    this.val = val === undefined ? null : val;
 *    this.left = left === undefined ? null : left;
 *    this.right = right === undefined ? null : right;
 *    this.next = next === undefined ? null : next;
 * };
 */

/**
 * @param {Node} root
 * @return {Node}
 */
var connect = function(root) {
    if(!root){
        return root
    }
    if(root.left){
        root.left.next = root.right
        root.right.next = root.next ? root.next.left : null
        connect(root.left)
        connect(root.right)
    }
    return root
}

结果：

• 执行结果： 通过
• 执行用时：88 ms, 在所有 JavaScript 提交中击败了35.93%的用户
• 内存消耗：47 MB, 在所有 JavaScript 提交中击败了58.94%的用户
• 通过测试用例：59 / 59

END