题目:二叉搜索树与双向链表
题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
解题思路:
ps:来自剑指 offer评论区
方法二:递归版
解题思路:
1.将左子树构造成双链表,并返回链表头节点。
2.定位至左子树双链表最后一个节点。
3.如果左子树链表不为空的话,将当前root追加到左子树链表。
4.将右子树构造成双链表,并返回链表头节点。
5.如果右子树链表不为空的话,将该链表追加到root节点之后。
6.根据左子树链表是否为空确定返回的节点。
代码实现:
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
import java.util.*;
public class Solution {
public TreeNode Convert(TreeNode pRootOfTree) {
// 代码的鲁棒性
if (pRootOfTree == null) {
return null;
}
if (pRootOfTree.left == null && pRootOfTree.right == null) {
return pRootOfTree;
}
// 将左子树调整为双向链表
TreeNode left = Convert(pRootOfTree.left);
TreeNode le = left;
// 左子树不为空: 定位至左链表的最后一个元素
while (le != null && le.right != null) {
le = le.right;
}
if (left != null) {
le.right = pRootOfTree;
pRootOfTree.left = le;
}
TreeNode right = Convert(pRootOfTree.right);
TreeNode rig=right;
// 左子树不为空: 定位至左链表的最后一个元素
if (right != null) {
right.left = pRootOfTree;
pRootOfTree.right = right;
}
return left!=null?left:pRootOfTree;
}
}