leetcode 1028从先序遍历还原二叉树 迭代写法

public TreeNode recoverFromPreorder(String S) {
    //利用栈来保存结点, 新遍历到的结点肯定是某一子节点(根结点除外),由于是先序遍历
    //那么可能是上一个遍历结点的左孩子,要么是之前的已经遍历的右孩子
    LinkedList<TreeNode> stack = new LinkedList<>();
    int pos = 0;
    while(pos<S.length()){
        //首先拿到深度信息
        int depth = 0;
        while(pos<S.length() && S.charAt(pos) == '-'){
            depth++;
            pos++;
        }
        //再拿到结点信息
        int value = 0;
        while(pos<S.length() && S.charAt(pos) != '-'){
            value = value*10 + S.charAt(pos) - '0';
            pos++;
        }
        TreeNode node = new TreeNode(value);
        if(depth == stack.size()){
           if(!stack.isEmpty()){
                stack.peek().left = node;
           }
        }else{//此时说明栈顶节点不是其父结点应该栈顶节点一直出栈 utill 满足条件
        //用迭代模拟回溯
            while(stack.size() != depth){
                stack.pop();
            }
            stack.peek().right = node;
        }
        stack.push(node);

    }
    TreeNode root = null;
    while(!stack.isEmpty()){
        root = stack.pop();
    }
    return root;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值