题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
解题思路
对于每个非子节点来说,他们的左节点是左子树中的最大值,也就是左子树中的最右节点;他们的右节点是右子树中的最小值,也就是右子树中的最左节点。
对于每个子节点来说,他们的父节点就是对应的他们的左节点或者右节点。
最后,我们需要找到双向链表的头节点位置。
代码
# -*- 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