LeetCode107,117. 二叉树层序遍历2Golang版

LeetCode107. 二叉树层序遍历2Golang版

1. 问题描述

给定一个二叉树，返回其节点值自底向上的层序遍历。 （即按从叶子节点所在层到根节点所在的层，逐层从左向右遍历）

例如：
给定二叉树 [3,9,20,null,null,15,7],

3

/
9 20
/
15 7
返回其自底向上的层序遍历为：

[
[15,7],
[9,20],
[3]
]

2. 思路

二叉树层序遍历，再将res数组反转

3. 代码

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func levelOrderBottom(root *TreeNode) [][]int {
    var res [][]int
    if root == nil {
        return res
    } 

    queue := []*TreeNode{root}
    for len(queue) != 0 {
        length := len(queue)
        var layer []int
        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)
            }
            layer = append(layer, queue[0].Val)
            queue = queue[1:]
        }
        res = append(res, layer)
    }

    for i := 0; i < len(res) / 2; i++ {
        res[i], res[len(res) - i - 1] = res[len(res) - i - 1], res[i]
    }

    return res
}

LeetCode 117. 填充每个节点的下一个右侧节点指针 IIGolang版

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
}