1.题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。如下图所示
数据范围:输入二叉树的节点数 0≤n≤10000≤n≤1000,二叉树中每个节点的值 0≤val≤10000≤val≤1000
要求:空间复杂度O(1)O(1)(即在原树上操作),时间复杂度 O(n)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 树的形状如上图
2.解题思路
使用指针pre记录当前遍历结点的前一个结点,使用指针head记录双向链表的头结点。
当pre为空时,表示当前遍历到结点为该链表的头结点,令head = cur;head仅赋值一次;
当pre不为空时,表示当前遍历到的结点不是头结点,那么就需要进行连接,即当前结点的前驱结点为pre,pre的后继结点为当前结点。
然后令pre = cur;继续向后寻找下一个结点,按照中序遍历的顺序连接得到的双向链表即为一个有序的序列,中序遍历结束后,返回头结点即为所求。
3.代码实现
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 head = null;
TreeNode pre = null;
public TreeNode Convert(TreeNode pRootOfTree) {
inOrder(pRootOfTree);
return head;
}
public void inOrder(TreeNode cur) {
if (cur == null) {
return;
}
inOrder(cur.left);
if (pre != null) {
cur.left = pre;
pre.right = cur;
} else {
head = cur;
}
pre = cur;
inOrder(cur.right);
}
}