题目描述:输入一颗二元查找树,将该二元查找树转换成一个排序的双向链表。要求不能有新的结点,只能调整结点的指向。
分析解答:由于转换后的双向链表中结点的顺序与二叉树的中序遍历相同,因此,可以对二叉树的中序遍历算法进行修改,通过在中序遍历的过程中修改结点的指向来
class BiTNode:
def __init__(self):
self.data = None
self.lchild = None
self.rchild =None
class Test:
def __init__(self):
self.pHead = None # 双向链表头结点
self.pEnd = None # 双向链表尾结点
# 把有序数组转换成二叉树
def arraytotree(self, array, start, end):
root = None
if end >= start:
root = BiTNode()
mid = int((start + end + 1)/2)
# 树的根节点为数组中间的元素
root.data = array[mid]
# 递归的用左半部分数组构造root的左子树
root.lchild = self.arraytotree(array, start, mid-1)
# 递归用右半部分数组构造root的右子树
root.rchild = self.arraytotree(array, mid + 1, end)
else:
root = None
return root
# 把二叉树转换成双向列表
def inOrderBSTree(self, root):
if root == None:
return
# 转换root的左子树
self.inOrderBSTree(root.lchild)
# 使当前结点的左子树指向双向链表中的最后一个结点
root.lchild = self.pEnd
# 双向列表为空,当前遍历的结点为双向链表的头结点
if None == self.pEnd:
self.pHead = root
else:
self.pEnd.rchild = root
self.pEnd = root
self.inOrderBSTree(root.rchild)
转换成一个排序的双向链表。