题目
解析
这个题,我不会做。。。我想的方法非常的简单。就是在遍历二叉树的时候,用另一个二叉树来存储内容。但是这样肯定是不行的,因为没有办法拼接。所以,必须分成两个步骤进行。
反思:另一个点在于,我知道是用遍历,但是没有想好拼接方法。
题目中提到了展开的链表和二叉树的先序遍历相同,就应该想到是先遍历左子树,再遍历右子树。遍历完再进行拼接,如何拼接?
答案是通过递归进行拼接。
/**
* 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 flat = (root)=>{
// 边界条件
if(!root) return null;
// 遍历左右子树,拉平左右子树
flat(root.left)
flat(root.right);
// 进行拼接
let left = root.left;
let right = root.right;
// 左子树为空
root.left = null;
// 将左子树绑定到右子树上
root.right = left;
// 遍历现在的右子树到最后一个节点
while(root.right) {
root = root.right;
}
// 将原本的右子树接到节点尾部
root.right = right;
}
return flat(root);
};
正确代码如上,为什么自己写不出来? 因为逻辑上还不够清晰。
总结
在写多了二叉树的题目之后,我们知道一定会对左右子树进行遍历,那么遍历结束之后,如何对左右子树进行处理呢? 是在遍历子树前进行处理,还是遍历子树后进行处理?
要有相关的思考才能进行操作。
可参考labuladong对于二叉树的分析。