原题如下:leetcode.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]
方法1:
通过深度优先遍历,不断将二叉树拉平,
- 例如上图,通过创建一个数组,不断向头部塞节点,从而实现后序排列,先塞右节点,再塞左节点
- 遍历是,不断从头部取节点,然后将当前节点,不断拼接在目标节点的尾部
- 例如 2 拼接在 1 后面,形成 1 > 2,指针指向 2
- 然后 3 拼接在 当前指针后面,指针目前指向 2,所以形成 1 > 2 > 3, 指针指向尾节点 3
- 右节点拼接在 当前指针后面,指针指向3,所以 1 > 2 > 3 > 4
- 依次类推,完成拼接操作
- 如果节点有子节点,则重复1,2操作
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root
* @return {void} Do not return anything, modify root in-place instead.
*/
var flatten = function(root) {
let newNode = new TreeNode(0);
const result = newNode;
const list = [];
if (root) {
list.push(root);
}
while(list.length) {
const item = list.shift();
if (item.right) {
list.unshift(item.right);
}
if (item.left) {
list.unshift(item.left);
}
newNode.right = new TreeNode(item.val);
newNode = newNode.right;
}
if (result.right) {
root.right = result.right.right;
root.left = null;
}
};
方法2(递归):
- 从最左侧的最底部开始向上遍历
- 将 1 < 2 > 3 -> 2 > 1, (当前节点的左节点,直接替换掉右节点)
- 从2开始向下查找, 找到1,1没有右节点,将2的旧右节点接在1后面,形成 2 > 1 > 3
- 这样一个节点就完成了,依次类推
- 2节点上还有父节点,例如 2 < 5 > 4
- 需要将左节点2全部拼接在父节点5上,也就是 5 > 2 > 1 > 3
- 依次向右遍历节点,发现3右侧没有节点,此时将原来5的右节点,拼接在3后面,
- 即可得到新的树, 5 > 2 > 1 > 3 > 4
- 依次向上传递,即可获得一个完整的单链表数
- 此方法,通过递归的形式,从最小节点开始拉平,慢慢拉平到全局,由点辐射到面
/**
* @param {TreeNode} root
* @return {void} Do not return anything, modify root in-place instead.
*/
var flatten = function(root) {
if (!root) return;
flatten(root.left);
flatten(root.right);
// 拉平一个节点
const right = root.right;
root.right = root.left;
root.left = null;
let p = root;
while (p.right) {
p = p.right;
}
p.right = right;
};