剑指offer(7)二叉搜索树与双向链表--python

题目描述
输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点，只能调整树中结点指针的指向。

非递归方法
思路：首先，二叉树搜索树的中序遍历是一个排序的遍历方法，并且，二叉树的中序遍历方法必须知道，不论是递归方法还是非递归的方法，都要很熟。
1.实现二叉搜索树的中序排列
2.将中序排列列表中的各个节点前后连接起来

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#非递归的方法：先中序遍历二叉树，对得到的序列中的节点左右连接
class Solution:
    def Convert(self, pRootOfTree):
        if not pRootOfTree:
            return None
         
        p = pRootOfTree
         
        stack = []
        resStack = []
        #中序遍历
        while p or stack:
            if p:
                stack.append(p)
                p = p.left
            else:
                node = stack.pop()#需要将最后一个节点pop出来，以进行下一步的对右子节点的遍历
                resStack.append(node)
                p = node.right
        #连接前后列表
        resP = resStack[0]
        while resStack:
            top = resStack.pop(0)
            if resStack:
                top.right = resStack[0]
                resStack[0].left = top
         
        return resP

递归方法

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def Convert(self, root):
        if not root:
            return None
        if not root.left and not root.right:
            return root
         
        # 将左子树构建成双链表，返回链表头
        left = self.Convert(root.left)
        p = left
         
        # 定位至左子树的最右的一个结点
        while left and p.right:
            p = p.right
         
        # 如果左子树不为空，将当前root加到左子树链表
        if left:
            p.right = root
            root.left = p
         
        # 将右子树构造成双链表，返回链表头
        right = self.Convert(root.right)
        # 如果右子树不为空，将该链表追加到root结点之后
        if right:
            right.left = root
            root.right = right
             
        return left if left else root

代码参考牛客网Yannyezixin