LeetCode116. 填充每个节点的下一个右侧节点指针Golang版

LeetCode116. 填充每个节点的下一个右侧节点指针Golang版

1. 问题描述

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

struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。

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

进阶：

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

2. 思路

二叉树层序遍历
每一层，如果不是最后一个节点，queue[0].Next = queue[1]
如果是最后一个节点，queue[0].Next = nil

3. 代码

/**
 * Definition for a Node.
 * type Node struct {
 *     Val int
 *     Left *Node
 *     Right *Node
 *     Next *Node
 * }
 */

func connect(root *Node) *Node {
	if root == nil {
        return nil
    }

    queue := []*Node{root}

    for len(queue) > 0 {
        length := len(queue)
        for i := 0; i < length; i++ {
            if queue[0].Left != nil {
                queue = append(queue, queue[0].Left)
            }
            if queue[0].Right != nil {
                queue = append(queue, queue[0].Right)
            }
            
            if i != length - 1 {
                queue[0].Next = queue[1]
            } else {
                queue[0].Next = nil
            }
            queue = queue[1:]
        }
    }

    return root
}