算法题-二叉搜索树与双向链表

[编程题]二叉搜索树与双向链表
时间限制：1秒空间限制：32768K

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

思路1：
非递归版本：
二叉树的中序遍历
中序遍历中每个结点的链接

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def Convert(self, pTreeRoot):
        if not pTreeRoot: #空树
            return None
        if pTreeRoot.left is None and pTreeRoot.right is None:#只有根节点
            return pTreeRoot
        # 递归将左子树构建成双链表，返回链表头
        left = self.Convert(pTreeRoot.left)
        Pointer = left
        # 定位左子树最右边节点
        while left and Pointer.right:
            Pointer = Pointer.right
             # 如果左子树不为空，将当前pTreeRoot加到左子树链表
        if left is not None:
            Pointer.right = pTreeRoot
            pTreeRoot.left = Pointer
             # 将右子树递归构造成双链表，返回链表头
        right = self.Convert(pTreeRoot.right)
        # 如果右子树不为空，将该链表追加到pTreeRoot结点之后
        if right is not None :
            right.left = pTreeRoot
            pTreeRoot.right = right    

        return left if left else pTreeRoot

思路2:python solution:
先中序遍历，将所有的节点保存到一个列表中。对这个list[:-1]进行遍历，每个节点的right设为下一个节点，下一个节点的left设为上一个节点。

class Solution:
    def Convert(self, pRootOfTree):
        # write code here
        if not pRootOfTree:return
        self.arr = []
        self.midTraversal(pRootOfTree)
        for i,v in enumerate(self.arr[:-1]):
            v.right = self.arr[i + 1]
            self.arr[i + 1].left = v
        return self.arr[0]

    def midTraversal(self, root):
        if not root: return
        self.midTraversal(root.left)
        self.arr.append(root)
        self.midTraversal(root.right)