文章目录
二叉搜索树与双向链表
描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。如下图所示
数据范围:输入二叉树的节点数 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 树的形状如上图
答案
/** public class TreeNode { int val = 0; TreeNode left = null; TreeNode right = null; public TreeNode(int val) { this.val = val; } } */ public class Solution { //返回的第一个指针,即为最小值,先定为null fast-template public TreeNode head = null; //中序遍历当前值的上一位,初值为最小值,先定为null public TreeNode pre = null; public TreeNode Convert(TreeNode pRootOfTree) { return LDR(pRootOfTree); } public TreeNode LDR(TreeNode root){ if(root == null){ //中序递归,叶子为空则返回 return null; } //首先递归到最左最小值 LDR(root.left); //找到最小值,初始化head和pre if(pre==null){ head = root; pre = root; }else{ //当前结点与上一结点建立连接,将pre设置为当前值 pre.right = root; root.left = pre; pre = root; } LDR(root.right); return head; } }