题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路:采用递归的思想。由于搜索二叉树的左子树的值都小于根节点,右子树的值都大于根节点,因此选择中序遍历,这样可以满足从大到小的顺序遍历每个节点的要求。当遍历到根节点的时候,左子树已经转换成一个排序的链表,链表的最后一个是最大的节点,将其与根节点连接起来。然后去转换右子树,右子树的最小的节点和根节点相连。
# -*- 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.right = left, pRootOfTree
#处理右子树
self.Convert(pRootOfTree.right)
right = pRootOfTree.right
#连接根节点与右子树最小节点
if right:
while right.left:
right = right.left
pRootOfTree.right, right.left = right, pRootOfTree
#全部连接完以后需要找到头节点
while pRootOfTree.left:
pRootOfTree = pRootOfTree.left
return pRootOfTree