【题目】
二叉树数据结构TreeNode可用来表示单向链表(其中left置空,right为下一个链表节点)。实现一个方法,把二叉搜索树转换为单向链表,要求依然符合二叉搜索树的性质,转换操作应是原址的,也就是在原始的二叉搜索树上直接修改。
返回转换后的单向链表的头节点。
注意:本题相对原题稍作改动
示例:
输入: [4,2,5,1,3,null,6,0]
输出: [0,null,1,null,2,null,3,null,4,null,5,null,6]
提示:
节点数量不会超过 100000。
【代码】
【迭代】
class Solution:
def convertBiNode(self, root: TreeNode) -> TreeNode:
if not root:
return root
stack=[]
cur=None
lastHead=None
while stack or root:
while root:
stack.append(root)
root=root.left
root=stack.pop()
if not cur:
cur=lastHead=root
else:
cur.right=root
cur=root
root.left=None
root=root.right
return lastHead
【递归】
class Solution:
def visit(self,root):
if not root:
return
self.visit(root.left)
if not self.cur:
self.cur=self.lastNode=root
else:
self.cur.right=root
self.cur=root
root.left=None
self.visit(root.right)
def convertBiNode(self, root: TreeNode) -> TreeNode:
self.lastNode=None
self.cur=None
self.visit(root)
return self.lastNode