2020.05.13
题目
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
点击链接
比如输入图4.12 中左边的二叉搜索树,则输出转换之后的排序现向链表。
结点定义
public static class BinaryTreeNode {
int value;
BinaryTreeNode left;
BinaryTreeNode right;
}
解题思路
在二叉树中,每个结点都有两个指向子结点的指针。在双向链表中,每个结点也有两个指针,它们分别指向前一个结点和后一个结点。由于这两种结点的结构相似,同时二叉搜索树也是一种排序的数据结构,因此在理论上有可能实现二叉搜索树和排序的双向链表的转换。在搜索二叉树中,左子结点的值总是小于父结点的值,右子结点的值总是大于父结点的值。因此我们在转换成排序双向链表时,原先指向左子结点的指针调整为链表中指向前一个结点的指针,原先指向右子结点的指针调整为链表中指向后一个结点指针。接下来我们考虑该如何转换。
由于要求转换之后的链表是排好序的,我们可以中序遍历树中的每一个结点, 这是因为中序遍历算法的特点是按照从小到大的顺序遍历二叉树的每一个结点。当遍历到根结点的时候,我们把树看成三部分:值为10 的结点、根结点值为6 的左子树、根结点值为1 4 的右子树。根据排序链表的定义,值为10 的结点将和它的左子树的最大一个结点(即值为8 的结点)链接起来,同时它还将和右子树最小的结点(即值为12 的结点)链接起来,如图4.13 所示。
按照中序遍历的顺序, 当我们遍历转换到根结点(值为10 的结点)时,它的左子树已经转换成一个排序的链表了, 并且处在链表中的最后一个结点是当前值最大的结点。我们把值为8 的结点和根结点链接起来,此时链表中的最后一个结点就是10 了。接着我们去地历转换右子树, 并把根结点和右子树中最小的结点链接起来。至于怎么去转换它的左子树和右子树,由于遍历和转换过程是一样的,我们很自然地想到可以用递归。
用程序的思路来说,我们通过递归实现:
我们声明一个长度为1结点数组,因为要实现类似链表的功能,相当于声明一个链表,我们把这个下标为0的,当作链表的尾结点;
// 10
// / \
// 6 14
// /\ /\
// 4 8 12 16
我们以上面的二叉查找树为例
1、我们将树不断递归到最左子结点,如果为空我们将结点的左子结点(left)指向尾结点(null),相当于前驱指向null;如果尾结点为空,则尾结点后继结点(right)指向当前根结点,不为空则指向当前结点,递归右子树同理,然后返回上层结点。
2、上个遍历结束后,当前结点左子结点(前驱)指向上个结点,如果尾结点(上个结点)不为空,上个结点右子结点(后继)指向当前结点,递归右子树同理。
3、我们进入到右子树,当前结点的左子结点(前驱)指向上个结点,上个结点右子结点(后继)指向当前结点,递归返回上上层结点。
这样4 -> 6 -> 8就形成啦
此处我是拿左子树的局部说明的,希望便于理解
package offer27;
public class Test {
public static class BinaryTreeNode {
int value;
BinaryTreeNode left;
BinaryTreeNode right;
}
public static BinaryTreeNode convert(BinaryTreeNode root) {
BinaryTreeNode[] endBinaryTreeNode = new BinaryTreeNode[1];
doConvert(root, endBinaryTreeNode);
BinaryTreeNode headBinaryTreeNode = endBinaryTreeNode[0];
while (headBinaryTreeNode != null && headBinaryTreeNode.left != null) {
headBinaryTreeNode = headBinaryTreeNode.left;
}
return headBinaryTreeNode;
}
public static BinaryTreeNode[] doConvert(BinaryTreeNode node,BinaryTreeNode[] endNode) {
if(node != null) {
if(node.left != null) {
doConvert(node.left,endNode);
}
node.left = endNode[0];
//如果左子树转换成的双向链表不为空,设置尾结点的后继
if (endNode[0] != null) {
endNode[0].right = node;
}
endNode[0] = node;
if(node.right != null) {
doConvert(node.right, endNode);
}
}
return endNode;
}
//二叉树中序遍历
private static void printTree(BinaryTreeNode root) {
if (root != null) {
printTree(root.left);
System.out.print(root.value + "->");
printTree(root.right);
}
}
//链表遍历
private static void printList(BinaryTreeNode head) {
while (head != null) {
System.out.print(head.value + "->");
head = head.right;
}
}
// 10
// / \
// 6 14
// /\ /\
// 4 8 12 16
private static void test01() {
BinaryTreeNode node10 = new BinaryTreeNode();
node10.value = 10;
BinaryTreeNode node6 = new BinaryTreeNode();
node6.value = 6;
BinaryTreeNode node14 = new BinaryTreeNode();
node14.value = 14;
BinaryTreeNode node4 = new BinaryTreeNode();
node4.value = 4;
BinaryTreeNode node8 = new BinaryTreeNode();
node8.value = 8;
BinaryTreeNode node12 = new BinaryTreeNode();
node12.value = 12;
BinaryTreeNode node16 = new BinaryTreeNode();
node16.value = 16;
node10.left = node6;
node10.right = node14;
node6.left = node4;
node6.right = node8;
node14.left = node12;
node14.right = node16;
System.out.print("Before convert: ");
printTree(node10);
System.out.println();
BinaryTreeNode head = convert(node10);
System.out.print("After convert : ");
printList(head);
System.out.println();
}
public static void main(String[] args) {
test01();
}
}