题目
给你二叉树的根结点 root
,请你将它展开为一个单链表:
- 展开后的单链表应该同样使用
TreeNode
,其中right
子指针指向链表中下一个结点,而左子指针始终为null
。 - 展开后的单链表应该与二叉树 先序遍历 顺序相同。
示例 1:
输入:root = [1,2,5,3,4,null,6]
输出:[1,null,2,null,3,null,4,null,5,null,6]
1
2
2
示例 2:
输入:root = []
输出:[]
1
2
2
示例 3:
输入:root = [0]
输出:[0]
1
2
2
提示:
- 树中结点数在范围
[0, 2000]
内 -100 <= Node.val <= 100
进阶: 你可以使用原地算法(O(1)
额外空间)展开这棵树吗?
题解
java
public void flatten(TreeNode root) {
if (root == null) {
return;
}
TreeNode left = root.left;
TreeNode right = root.right;
root.left = null;
root.right = null;
// 递归左树
this.flatten(left);
// 递归右树
this.flatten(right);
// 左树直接拼接到右树
if (null != left) {
root.right = left;
}
// 若右树存在 拼接在当前右树后面
if (null != right) {
TreeNode cursor = root;
while (cursor.right != null) {
cursor = cursor.right;
}
cursor.right = right;
}
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25