使用栈来模拟递归过程实现二叉树的遍历

二叉树的定义代码

          public class TreeNode {
              int val;
              TreeNode left;
              TreeNode right;
              TreeNode(int x) { val = x; }
          }

中序遍历代码

    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        while(stack.size()>0 || root!=null){
            if(root!=null){
                stack.add(root);
                root = root.left;
            }else{
                TreeNode t = stack.pop();
                res.add(t.val);
                root = t.right;
            }
        }
        return res;
    }

中序遍历用递归的调用过程是不断往左边走,当左边走不下去了,就打印节点,并转向右边,然后右边继续这个过程。

首先传入一个有值的二叉树节点,开始执行循环,如果当前节点非空,则将当前整个节点压入栈中,注意是整个节点而不是当前节点的val。然后用该节点的左节点,替换传入的节点,如此循环,一直到节点为null,则表示到了树的最底端,这时候就可以把原先压入到栈里的元素逐个取出,然后将节点的值添加到结果集中,然后转向右边,重复上述过程,最后输出的结果集就是中序遍历的结果

前序遍历代码

    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        while(stack.size()>0 || root!=null){
           if(root!=null){
                stack.add(root);
                res.add(root.val);
                root = root.left;
           }else{
               TreeNode t =stack.pop();
               root = t.right;
           }      
        }
        return res;
    }

只需要将res.add移动到访问左节点之前,即可实现前序遍历,后序同理。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值