105. Construct Binary Tree from Preorder and Inorder Traversal
题目意思:已知先序和中序,建立一棵二叉树。
这篇文章之前在这里已经分享过了,值得好好学习。
地址:https://blog.csdn.net/sir_TI/article/details/89277987
114. Flatten Binary Tree to Linked List
题目分析:给定一棵二叉树,将其原地展开为一个链表。先看题目给出的例子,然后再分析:
Example:
给出如下的二叉树:
1
/ \
2 5
/ \ \
3 4 6
经过展开(flatten)以后,其变成:
1
\
2
\
3
\
4
\
5
\
6
结合例子来看,本质的意思就是取出先序序列,然后按照先序序列的顺序建树,要求这棵树的左孩子结点均为NULL,右孩子结点是其先序序列的下一个结点。
题目解答:弄懂了题目的意思,分析可得,一般情况下,我们可以按照某种方式去遍历,最终取得最后的结果,在本题中,我们可以按照后序遍历的思想,注意这里有点不同的是,我们是先平整右子树,再平整左子树,最后来处理根节点,这里平整的时候借助于一个前驱,保证不断链。
class Solution:
def __init__(self):
self.prev = None
def flatten(self, root):
if not root:
return None
# 平整右边
self.flatten(root.right)
# 平整左边
self.flatten(root.left)
# 修改当前节点的右孩子指针指向其前驱
root.right = self.prev
# 当前节点的右孩子指向None
root.left = None
self.prev = root