题目
给你二叉树的根结点 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 = []
输出:[]
题解
const flatten = (root) => {
const helper = (root) => { // 将当前子树转成一个单链表
if (root == null) return null; // 遍历到null节点 返回null节点
if (root.right) { // 先生成右子树的单链表
helper(root.right);
}
if (root.left) { // 如果有左子树,生成单链表然后搬运过去
const leftFirst = helper(root.left); // 生成单链表,并获取头结点
//此时就默认左链表是一个向右倾斜的单链表了
let leftLast = leftFirst; // leftEnd是单链表的尾节点
while (leftLast.right) { // 一直找右节点,获取到单链表的尾节点
leftLast = leftLast.right;
}
leftLast.right = root.right; // 尾节点后面接左子树展平后的单链表
//根节点是最小的,所以根节点不动,把左子树变成的链表插到根和右链表之间
root.right = leftFirst; // 根节点的right改成leftFirst
root.left = null; // root.left置为null
}
return root; // 返回出当前子树转成的单链表
};
helper(root); // 原地修改,不用返回
};
笔记:
-
具体思路
右子树转成的单链表,只要获取它的头结点,等左子树生成的链表生成好,接到它的尾节点
左子树生成的单链表,处理稍微复杂一点- 要获取它的头结点,它要接到根节点的right
- 要获取它的尾节点,通过一直找右节点,找到尾节点,供连接
左子树生成的链表两端都接好后,root.left 要置为 null,不然 root 还拖着个左子树