题目描述:
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。如下图所示:
数据范围:输入二叉树的节点数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;
解法:迭代法
思路:
借助中序遍历的非递归写法,在中序遍历的过程中通过修改当前遍历节点与前一遍历节点的指针指向。
1、cur指向当前节点,定义pre指向cur的前一节点。
2、按照中序遍历的特点,先遍历左子树,然后修改为双向链表,然后遍历右子树
代码:
/**
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 {
TreeNode pre = null, head = null;
public TreeNode Convert(TreeNode root) {
//处理节点为空的情况
if(root == null){
return null;
}
//辅助栈
Stack<TreeNode> stack = new Stack<>();
//指针
TreeNode cur = root;
//记录前一节点
TreeNode pre = null;
boolean isFirst = true;
while(cur != null || !stack.isEmpty()){
//将左节点入栈
while(cur != null){
stack.push(cur);
cur = cur.left;
}
//取出栈顶元素
cur = stack.pop();
if(isFirst){ //将最左节点作为返回值root
root = cur;
pre = root;
isFirst = false;
}else{
//将前一节点的右指针指向当前节点
pre.right = cur;
//当前节点的左指针指向前一节点
cur.left = pre;
//更改pre指向
pre = cur;
}
//指向右子节点
cur = cur.right;
}
return root;
}
}