剑指offer算法题
分治 二叉搜索树和链表
题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
题目分析
-
明确Convert函数的功能。
输入:输入一个二叉搜索树的根节点。
过程:将其转化为一个有序的双向链表。
输出:返回该链表的头节点。 -
明确成员变量pLast的功能。
pLast用于记录当前链表的末尾节点。 -
明确递归过程。
递归的过程就相当于按照中序遍历,将整个树分解成了无数的小树,然后将他们分别转化成了一小段一小段的双向链表。再利用pLast记录总的链表的末尾,然后将这些小段链表一个接一个地加到末尾。
每次递归过程中,首先将当前节点的原左子树指针,指向所储存的上一个节点pLast;之后将前一个节点pLast的原右子树指针,指向当前节点。如此,完成了一个节点的链表转换。
下面是JAVA算法实现:
private TreeNode pLast = null;
public TreeNode Convert(TreeNode pRootOfTree) {
if(pRootOfTree == null){
return null;
}
// 如果左子树为空,那么根节点root为双向链表的头节点
TreeNode head = Convert(pRootOfTree.left);
if(head ==null){
head = pRootOfTree;
}
// 连接当前节点root和当前链表的尾节点pLast
pRootOfTree.left = pLast;
if(pLast!=null){
pLast.right = pRootOfTree;
}
pLast = pRootOfTree;
Convert(pRootOfTree.right);
return head;
}