[JZoffer]二叉搜索树与双向链表

题目描述

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

解题思路

对于每个非子节点来说，他们的左节点是左子树中的最大值，也就是左子树中的最右节点；他们的右节点是右子树中的最小值，也就是右子树中的最左节点。
对于每个子节点来说，他们的父节点就是对应的他们的左节点或者右节点。
最后，我们需要找到双向链表的头节点位置。

代码

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def Convert(self, pRootOfTree):
        # write code here
        if not pRootOfTree:
            return pRootOfTree
        if not pRootOfTree.left and not pRootOfTree.right:
            return pRootOfTree
        #处理左子树
        self.Convert(pRootOfTree.left)
        left = pRootOfTree.left
        #连接根与左子树最大节点
        if left:
            while left.right:  # 左子树有右节点的话那么就一直去找寻左子树的最右节点
                left = left.right
            pRootOfTree.left = left
            left.right = pRootOfTree

        #处理右子树
        self.Convert(pRootOfTree.right)
        right = pRootOfTree.right
        #连接根与右子树最小节点
        if right:
            while right.left:  # 右子树有左节点的话那么就一直去找寻右子树的最左节点
                right = right.left
            pRootOfTree.right = right
            right.left = pRootOfTree

        while pRootOfTree.left:  # 找到双向链表的头节点位置
            pRootOfTree = pRootOfTree.left
        return pRootOfTree