题目描述

给你二叉树的根结点 root ，请你将它展开为一个单链表：

• 展开后的单链表应该同样使用 TreeNode ，其中 right 子指针指向链表中下一个结点，而左子指针始终为 null 。
• 展开后的单链表应该与二叉树 先序遍历 顺序相同。

示例 1：

输入： root = [1,2,5,3,4,null,6]
输出： [1,null,2,null,3,null,4,null,5,null,6]

示例 2：

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

示例 3：

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

提示：

• 树中结点数在范围 [0, 2000] 内
• -100 <= Node.val <= 100

代码及注释

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */

func flatten(root *TreeNode)  {
    // 初始化一个用于存储二叉树节点的列表和一个栈
    var list, stack []*TreeNode
    // 迭代遍历二叉树，直到栈为空且当前节点为nil
    for len(stack) > 0 || root != nil {
        // 将当前节点及其左子节点加入列表和栈中
        for root != nil {
            list = append(list, root)
            stack = append(stack, root)
            root = root.Left
        }
        // 当前节点为栈顶节点的右子节点
        root = stack[len(stack) - 1].Right
        // 弹出栈顶节点
        stack = stack[:len(stack) - 1]
    }

    // 遍历节点列表，将每个节点的左子节点设为nil，右子节点设为下一个节点
    for i := 1; i < len(list); i++ {
        prev, curr := list[i-1], list[i]
        prev.Left, prev.Right = nil, curr
    }
}

代码解释

• 使用先序遍历遍历二叉树，同时使用一个栈来存储遍历的节点。
• 遍历结束后，节点列表list中存储了二叉树的所有节点，并且是按照先序遍历的顺序。
• 再次遍历list，将每个节点的左子节点设为nil，右子节点设为下一个节点，即可将二叉树展开为单链表。