给你二叉树的根结点
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
进阶:你可以使用原地算法(
O(1)
额外空间)展开这棵树吗?
思路:
- 先前序遍历得到所有节点
- 然后再拼接成链表
时间复杂度:O(N)
空间复杂度:O(N)
/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
// 1、先前序遍历得到所有节点,保存到切片中
// 2、然后再拼接成链表
func flatten(root *TreeNode) {
preOrderList := make([]*TreeNode, 0)
var dfs func(node *TreeNode)
dfs = func(node *TreeNode) {
if node == nil {
return
}
preOrderList = append(preOrderList, node)
dfs(node.Left)
dfs(node.Right)
}
// 1、先前序遍历得到所有节点,保存到切片中
dfs(root)
// 2、然后再拼接成链表
for i := 0; i < len(preOrderList) - 1; i++ {
preOrderList[i].Right = preOrderList[i+1]
preOrderList[i].Left = nil
}
}