【树-简单】面试题 17.12. BiNode

【题目】
二叉树数据结构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