LeetCode 题解

积跬步，至千里。

二叉树展开为链表

分类：LeetCode 标签： 树 深度优先搜索 发布于：2019-10-20 ID：114

题目描述

来源于 https://leetcode-cn.com/

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

例如，给定二叉树

    1
   / \
  2   5
 / \   \
3   4   6

将其展开为：

1
 \
  2
   \
    3
     \
      4
       \
        5
         \
          6

解法：

先序遍历，始终记录下前一个节点，然后将当前节点接在前一个节点的右孩子上。

为了避免初始阶段 last 为空（即开始的时候前一个节点是不存在的），可以随意给 last 初始化一个节点，这并不影响结果。

class Solution:
    def flatten(self, root: TreeNode) -> None:
        """
        Do not return anything, modify root in-place instead.
        """
        
        stack = [root]
        last = TreeNode(0)
        
        while stack:
            node = stack.pop()
            if not node:
                continue
        
            last.right = node
            last = node

            stack.append(node.right)
            stack.append(node.left)
            node.left = None