题目描述
题目链接:二叉搜索树与双向链表_牛客题霸_牛客网 (nowcoder.com)
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。如下图所示
数据范围:输入二叉树的节点数 0≤n≤1000,二叉树中每个节点的值 0≤val≤1000
要求:空间复杂度O(1)(即在原树上操作),时间复杂度O(n)注意:
1.要求不能创建任何新的结点,只能调整树中结点指针的指向。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继
2.返回链表中的第一个节点的指针
3.函数返回的TreeNode,有左右指针,其实可以看成一个双向链表的数据结构4.你不用输出双向链表,程序会根据你的返回值自动打印输出
输入描述:
二叉树的根节点
返回值描述:
双向链表的其中一个头节点。
示例1
输入:
{10,6,14,4,8,12,16}返回值:
From left to right are:4,6,8,10,12,14,16;From right to left are:16,14,12,10,8,6,4;说明:
输入题面图中二叉树,输出的时候将双向链表的头节点返回即可。
示例2
输入:
{5,4,#,3,#,2,#,1}返回值:
From left to right are:1,2,3,4,5;From right to left are:5,4,3,2,1;说明:
5 / 4 / 3 / 2 / 1 树的形状如上图
题目解读:
题目话不多,补充的要求多,难度是真的狠,属于是人狠话不多,社会我牛客(ge)。总结一句话就是,把搜索二叉树转为双向链表,不能借用其他数据结构,返回值最小的节点。 读完有点头皮发麻。
解题思想:
要解题,那就得了解搜索二叉树的特点,它是任意节点大于左子树,并且大于它的所有左子树(左边的儿子,孙子都比它小),同时 小于它的右子树,并且小于它的所有右子树。
所以通过左递归,可以找到最小值,右递归找到最大值。那就先进行左递归,当节点为空时,说明它前一个节点就是最小值节点,也就是要返回的头节点。如图所示,
当左递归到空时,它会返回上一层,上一层就是 4这个节点,然后把pre 和head都放在这个节点上,然后它再返回上一层,也就是6这个节点,这个时候,让pre.right 指向6 这个节点,6这个节点的 .left 指向pre ,然后pre移动到 6这个节点(因为题目要求,right指向后继节点,left指向 前驱节点)。 这里面要注意,利用递归回退,实现链表连接,前提是 head和 pre 都是全局节点,它们不受递归前进或回退的影响。 还有就是,第一个节点(也就是4这个节点),它比较特殊,它的前驱没有节点,所以对于它要单独处理。
在访问完6这个节点后,进行右递归,重复上面流程(4之后也有进行右递归,但是因为4的右子树为空,所以没说)。
代码注释:
import java.util.*;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
TreeNode pre = null;
TreeNode head = null;
public TreeNode Convert(TreeNode pRootOfTree) {
if(pRootOfTree == null)
return null;
// 先进行左递归
Convert(pRootOfTree.left);
// 找到最小值的节点后,就把pre和 head节点放上面
if(pre == null){
pre = pRootOfTree;
head = pRootOfTree;
// 最小值节点后,其他节点都是right连后继,left连前驱
}else{
pre.right = pRootOfTree;
pRootOfTree.left = pre;
pre = pRootOfTree;
}
// 当前节点遍历完后(也就是 左 根 右 的 根遍历完了),就遍历右子树
Convert(pRootOfTree.right);
return head;
}
}