[Leetcode][python]Flatten Binary Tree to Linked List/二叉树展开为链表
[Leetcode][python]Flatten Binary Tree to Linked List/二叉树展开为链表
蛮三刀酱
发布于 2019-03-26 17:05:15
发布于 2019-03-26 17:05:15
题目大意
把一棵二叉树变为链表（扁平化），也就是一棵所有节点要么没有子节点，要么只有右节点的二叉树。
解题思路
参考答案
思路：递归实现，暂存右结点，将左结点接在根结点右边，然后把暂存的右结点接在后面
可以看出来变化后每个节点其实都是指向了在先序遍历中的后一个节点。所以就通过栈的方式来先序遍历原树，如果一个节点有左节点，那么把它的右节点压栈（如果有的话），右指针指向原来的左节点；如果一个节点没有子节点，应该把它的右指针指向栈顶的节点。
代码
class Solution(object):
def flatten(self, root):
"""
:type root: TreeNode
:rtype: void Do not return anything, modify root in-place instead.
"""
pointer = dummy = TreeNode(None)
stack = [root]
while stack:
top = stack.pop()
if not top: continue
stack.append(top.right)
stack.append(top.left)
pointer.right = top
pointer.left = None
pointer = top总结
本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2017年08月04日，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录