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]
题解:我们可以发现展开的顺序就是二叉树的先序遍历(中左右),题目让我们把树的左子树插入到右子树中。
方法一、逐层遍历
思路:怎么理解这个逐层遍历呢?也就是遍历我们的树,把树的每一层的 左子树插入到我们这一层的右子树中,首先是找到我们左子树的最右边子节点,为什么呢?因为我们要插入把左子树插入到右子树中,就需要 把右子树接到我们左子树的最右边子节点 ,也就是 左子树的最右边子节点 = 右子树,这一层的左子树置 null ,我们重复这个过程就 ok 了。
图例:
1
/ \
2 5
/ \ \
3 4 6
左子树的最右边子节点 = 右子树
1
\
2
/ \
3 4
\
5
\
6
下一层 就是 2,那么就是
1
\
2
\
3
\
4
\
5
\
6
代码:
public void flatten(TreeNode root) {
while (root != null){
// 找到左子树的最右边的节点
if (root.left == null){
root = root.right;
}else {
TreeNode pre = root.left;
while (pre.right != null){
pre = pre.right;
}
//让右子树与其连接
pre.right = root.right;
//将左子树插入到右子树中
root.right = root.left;
//左子树置 null
root.left = null;
//进入下一个节点
root = root.right;
}
}
}
方法二、DFS
public void flatten(TreeNode root) {
dfs(root);
}
private TreeNode dfs(TreeNode root) {
if (root == null || root.left == null && root.right == null) {
return root;
}
TreeNode left = dfs(root.left);
TreeNode right = dfs(root.right);
if (left != null) {
//左子树的最右边子节点 = 右子树
left.right = root.right;
//将左子树插入到右子树
root.right = root.left;
root.left = null;
}
return right == null ? left : right;
}