给你二叉树的根结点 root ,请你将它展开为一个单链表:
- 展开后的单链表应该同样使用 TreeNode ,其中 right 子指针指向链表中下一个结点,而左子指针始终为 null 。
- 展开后的单链表应该与二叉树 先序遍历 顺序相同。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list
题目解读:就是将左子树和右子树展开为链表之后,先将左子树放到根节点的右边再将右子树放入原左子树的右边
这样的话就是用递归来处理
# Definition for a binary tree node. # class TreeNode: # def __init__(self, val=0, left=None, right=None): # self.val = val # self.left = left # self.right = right class Solution: def flatten(self, root: TreeNode) -> None: """ Do not return anything, modify root in-place instead. """ if not root: return self.flatten(root.left) self.flatten(root.right) # 将root右子树暂存 temp = root.right # 左子树放到右边 root.right = root.left # 将左子树置空 root.left = None # 找到原左子树的最右边 while root.right: root = root.right root.right = temp