题目：原题链接（中等）

标签：树、二叉树、深度优先搜索、栈

解法一（栈）：

class Solution:
    def flatten(self, root: TreeNode) -> None:
        stack = []
        node = root
        while node:
            # 处理节点同时包含左右节点的情况
            if node.left and node.right:
                stack.append(node.right)
                node.left, node.right = None, node.left

            # 处理节点只包含左节点的情况
            elif node.left:
                node.left, node.right = None, node.left

            # 处理节点既没有左节点也没有右节点的情况（取出栈顶元素左右右节点）
            elif not node.right and stack:
                node.right = stack.pop()

            node = node.right

解法二（不使用栈存储，直接放到对应的树的位置上）

class Solution:
    def flatten(self, root: TreeNode) -> None:
        node = root
        while node:
            if node.left:
                # 寻找当前右子树应移动到的位置
                now = node.left
                while now.right:
                    now = now.right

                # 移动当前右子树
                now.right = node.right

                # 修改当前节点
                node.left, node.right = None, node.left

            node = node.right