中序遍历DFS2.二叉搜索树与双向链表之间的转换

二叉搜索树和双向链表之间的转换

这个题需要将二叉排序树中的节点,变成按顺序排序的双向链表。因为双向链表和树的每个节点都具有两个指针,而且都是排序的,所以可行。要做的就是把一个节点的一个指针指向前驱节点,一个指针指向后驱节点。当然了这个没有父节点,不能遍历有限的次数找到前驱和后继,所以是中序遍历。

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);
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值