LeetCode50天刷题计划第二季（Day 13 — 填充每个节点的下一个右侧节点指针（16.00-16.30）-CSDN博客

本文链接：https://blog.csdn.net/weixin_46447549/article/details/127229247

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

文章目录

前言
一、题目
二、思路
三、代码

前言

困死了啊啊啊啊啊

一、题目

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

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

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 = []
输出：[]

提示：

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

进阶：

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

二、思路

对于每个节点依次处理其左孩子、右孩子即可很简单没得哈子困难

三、代码

"""
# Definition for a Node.
class Node:
    def __init__(self, val: int = 0, left: 'Node' = None, right: 'Node' = None, next: 'Node' = None):
        self.val = val
        self.left = left
        self.right = right
        self.next = next
"""

class Solution:
    def connect(self, root: 'Optional[Node]') -> 'Optional[Node]':
        if(root==None): #特殊情况
            return None
        head=root #head标记返回的根结点，root指向本层结点
        flag=root.left #flag标记下一层的第一个结点，对于每个结点， 如果有下一层，将其下一层的结点做好
        while(flag != None):
            while(root != None): #遍历本层结点
                root.left.next=root.right #处理当前结点左孩子
                if(root.next):
                    root.right.next=root.next.left #处理当前结点右孩子
                root=root.next
            root=flag #root到下一层
            flag=flag.left #flag到下一层
        return head

在这里插入图片描述