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,#]
解释：给定二叉树如图 A 所示，你的函数应该填充它的每个 next 指针，以指向其下一个右侧节点，如图 B 所示。序列化的输出按层序遍历排列，同一层节点由 next 指针连接，'#' 标志着每一层的结束。

例2：

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

二、解题思路

印象中做过类似的题，就是用广度优先搜索的方法，把每层的节点都压到一个队列里面，然后将每一个节点的next指针指向后一个。过程中维护两个队列，一个是当前队列，另一个是当前队列子节点构成的队列。但是这样写思路虽然简单，存在一个问题就是如果二叉树层数很高时，后期队列会很长，消耗存储空间大。注意到对于一个节点，左子节点的next总是指向右子节点，而右子节点的next指向旁边节点的左子节点，即当前节点next节点的左子节点。

三、AC代码

var connect = function(root) {
    if (root == null) return null;
    if (root.left != null) {
        root.left.next = root.right;
        if (root.next != null) root.right.next = root.next.left;
    }
    connect(root.left);
    connect(root.right);
    return root;
};

四、总结

总体来说，这道题的优化是要注意到上一层的next节点关系已经连好。