二叉搜索树和双向链表之间的转换
这个题需要将二叉排序树中的节点,变成按顺序排序的双向链表。因为双向链表和树的每个节点都具有两个指针,而且都是排序的,所以可行。要做的就是把一个节点的一个指针指向前驱节点,一个指针指向后驱节点。当然了这个没有父节点,不能遍历有限的次数找到前驱和后继,所以是中序遍历。
1.非递归版本
非递归版本很好写,首先按照正常的中序遍历非递归写出来,然后就是在到达最左端节点之后,也就是while循环到node为空之后进入else,这时要处理pop出的节点cur,最后让cur等于它的右节点,千万注意这里一定不用判断右节点是否为空,因为为空才能退出循环。
然后需要处理的就是第一步在外面设置一个flag来标志这是不是最左的节点,最左的节点就是第一个节点,令初始的root等于这个节点以便最后输出,并让flag变了。
下次进来这就不是第一个了,要处理的就是让pre的right等于当前的cur,让cur的left等于pre,这样就把cur和pre连起来了,最后令cur等于cur的right,继续循环。
import java.util.Stack;
public class Solution {
public TreeNode Convert(TreeNode pRootOfTree) {
if(pRootOfTree==null){
return null;
}
Stack<TreeNode> s=new Stack<TreeNode>();
boolean flag=true;//标志第一个节点好返回
TreeNode cur=pRootOfTree;//用于当前节点的运行,为什么不用pRootOfTree,因为pRootOfTree要返回,不能破坏到最后找不到
TreeNode pre=pRootOfTree;//用于保存上一个节点。
while(cur!=null||!s.empty()){
if(cur!=null){
s.push(cur);
cur=cur.left;
}else{
cur=s.pop();
if(flag){//如果是第一个节点,让root为它好返回
pRootOfTree=cur;
pre=pRootOfTree;
flag=false;
}else{//说明不是第一个节点了,处理上次节点的右和这次节点的左
pre.right=cur;
cur.left=pre;
pre=cur;
}
cur=cur.right;
}
}
return pRootOfTree;
}
}
2.递归
递归和普通的也一样,只不过因为每次都要返回一个值太麻烦,所以创建了一个新的函数。这里就是普通递归,然后在中间处理,处理的步骤 就两步,第一步是将cur的left指向pre,pre可以定义为全局变量。如果pre不为null,则设置它的right,然后就没了。最后还有一个特别巧妙的是设置一个全局的root为null,这样只在第一次进来的时候会设置它。
public class Solution {
TreeNode root=null;
TreeNode pre=null;
public TreeNode Convert(TreeNode pRootOfTree) {
if(pRootOfTree==null){
return null;
}
core(pRootOfTree);//为了不返回节点而设置函数
return root;
}
public void core(TreeNode pRootOfTree){
if(pRootOfTree==null){
return;
}
core(pRootOfTree.left);
pRootOfTree.left=pre;
if(pre!=null){
pre.right=pRootOfTree;
}
pre=pRootOfTree;
if(root==null){
root=pRootOfTree;
}
core(pRootOfTree.right);
}
}