题目描述:
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路:
二叉搜索树的中序遍历序列就是排序的数列;
然后将当前节点的右子树设为下一个节点,下一个节点的左子树设为该节点,即可。
(先中序遍历生成有序数列arr,之后进行左右节点设置)
python代码实现如下:
# -*- 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 None
self.arr = []
self.middle(pRootOfTree)
for i in range(len(self.arr)-1):
self.arr[i].right = self.arr[i+1]
self.arr[i+1].left = self.arr[i]
return self.arr[0]
def middle(self,root):
if not root:
return None
self.middle(root.left)
self.arr.append(root)
self.middle(root.right)