题目描述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
在线链接
代码:
import java.util.*;
public class Solution {
public TreeNode Convert(TreeNode pRootOfTree) {
if (pRootOfTree == null) {
return pRootOfTree;
}
Stack<TreeNode> st = new Stack<>();
TreeNode pre = null;
TreeNode p = pRootOfTree;
Boolean isFirst = true;
while (p != null || !st.isEmpty()) {
while (p != null){
st.add(p);
p = p.left;
}
p = st.pop();
if(isFirst){
pRootOfTree = p;
pre = p;
isFirst = false;
}else {
pre.right = p;
p.left = pre;
pre = p;
}
p = p.right;
}
return pRootOfTree;
}
}
思路概述:
要求将二叉搜索树转换成有序的链表,那肯定是和中序遍历有关
我们定义一个p指向root,定义一个pre记录p上一次到达的位置,定义一个isFirst来判断是否是第一次操作(因为中序遍历拿到的第一个结点会是链表的头结点,需要记录),如图是一开始的样子,蓝线代表左子树,红线代表右子树,方便区分。
中序遍历的操作不再赘述,第一次操作之后,变成如图
结束返回root即可。