题目:输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
思路:二叉搜索树的题目可以想到由递归来解决。由于要转换成排序的双链表,因此需要使用二叉树的前序遍历。先转换左子树,然后将根节点与左子树转换成的链表连接起来,再连接上右子树的转换链表即可。
代码实现:
// 返回排序双链表的头节点
public TreeNode Convert(TreeNode pRootOfTree) {
//robust
if(pRootOfTree == null)
return null;
//edge condition
if(pRootOfTree.left == null && pRootOfTree.right == null)
return pRootOfTree;
TreeNode head = null;
if(pRootOfTree.left != null){
head = Convert(pRootOfTree.left);
// 返回的是链表的头节点,所以需要移动到尾部
TreeNode leftNode = head;
while(leftNode.right != null)
leftNode = leftNode.right;
// 设置根节点的左右链接
leftNode.right = pRootOfTree;
pRootOfTree.left = leftNode;
}
if(pRootOfTree.right != null){
TreeNode rightNode = Convert(pRootOfTree.right);
pRootOfTree.right = rightNode;
rightNode.left = pRootOfTree;
}
return head!=null?head:pRootOfTree;
}