114_二叉树展开为链表

clbobo

于 2019-12-18 10:02:25 发布

阅读量91

点赞数

分类专栏： Leetcode

本文链接：https://blog.youkuaiyun.com/weixin_41529389/article/details/103592427

版权

Leetcode 专栏收录该内容

87 篇文章

订阅专栏

"""
给定一个二叉树，原地将它展开为链表。

例如，给定二叉树

    1
   / \
  2   5
 / \   \
3   4   6
将其展开为：
1
 \
  2
   \
    3
     \
      4
       \
        5
         \
          6
"""


def flatten(root):
    def helper(root):
        if root is None:
            return
        helper(root.left)
        helper(root.right)
        if root.left:
            pre = root.left          # 使用了辅助变量
            while pre.right:
                pre = pre.right
            pre.right = root.right
            root.right = root.left
            root.left = None
    helper(root)


def flatten2(root):
    while root:
        if root.left:
            most_right = root.left
            while most_right.right:
                most_right = most_right.right
            most_right.right = root.right
            root.right = root.left
            root.left = None
        root = root.right
    return