【力扣刷题】二叉树展开为链表

114.二叉树展开为链表
给你二叉树的根结点 root ，请你将它展开为一个单链表：
展开后的单链表应该同样使用 TreeNode ，其中 right 子指针指向链表中下一个结点，而左子指针始终为 null 。
展开后的单链表应该与二叉树 先序遍历 顺序相同。

先序遍历是肯定需要的，先序遍历又分为递归与迭代。先使用递归或迭代来求得先序遍历节点集合。再构建新的树。构建才是本题最关键的点。

递归代码：

class Solution(object):
    def flatten(self, root):
        """
        :type root: TreeNode
        :rtype: None Do not return anything, modify root in-place instead.
        """
        pre = []

        def PreOrder(root):
            if not root:
                return
            pre.append(root)
            PreOrder(root.left)
            PreOrder(root.right)
        # 先序遍历的递归
        PreOrder(root)
        for i in range(1, len(pre)):  # 必须从1开始，因为pre_node才是真正的当前节点
            pre_node, curr_node = pre[i-1], pre[i]  # 将前面的节点与当前节点找出来
            pre_node.left = None  # 将前面的节点作为当前节点，左子树置空
            pre_node.right = curr_node

迭代代码：

class Solution(object):
    def flatten(self, root):
        """
        :type root: TreeNode
        :rtype: None Do not return anything, modify root in-place instead.
        """
        pre = []

        stack = []
        p = root

        while stack or p:
            if p:
                pre.append(p)
                stack.append(p)
                p = p.left
            else:
                p = stack.pop()
                p = p.right
        
        for i in range(1, len(pre)):
            pre_node, curr_node = pre[i-1], pre[i]
            pre_node.left = None
            pre_node.right = curr_node