本文主要是介绍LeetCode-热题100:114. 二叉树展开为链表,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
题目描述
给你二叉树的根结点 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// 迭代遍历二叉树,直到栈为空且当前节点为nilfor 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
,右子节点设为下一个节点,即可将二叉树展开为单链表。
这篇关于LeetCode-热题100:114. 二叉树展开为链表的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!