二叉树前中后序遍历——非递归

import java.util.ArrayList;
import java.util.List;
import java.util.Stack;

class TreeNode
 {
     int val;
     TreeNode left;
     TreeNode right;
     TreeNode(int x) { val = x; }
 }
public class solution{
//前序(非递归)
    public List<Integer> preorderTraversal1(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        TreeNode curNode = root;
        while (!stack.isEmpty() || curNode != null) {
            while (curNode != null) {
                //访问根结点元素,并将根结点入栈
                list.add(curNode.val);
                stack.push(curNode);
                curNode = curNode.left;
            }
            //如果为空,取栈顶元素
            curNode = stack.pop();
            curNode = curNode.right;
        }
        return list;
    }

    //中序(非递归)
    public List<Integer> inorderTraversal(TreeNode root)
    {
        Stack<TreeNode> stack = new Stack<>();
        List<Integer> list = new ArrayList<>();
        TreeNode curNode = root;
        while(curNode != null || !stack.isEmpty()) {
            while (curNode != null) {
                //一路向左保存结点,直到为空
                stack.push(curNode);
                curNode = curNode.left;
            }
            //如果为空,取出栈顶元素,并访问栈顶元素
            curNode = stack.pop();
            list.add(curNode.val);
            curNode = curNode.right;
        }
        return list;
    }

    //后序(非递归)
    public List<Integer> postorderTraversal(TreeNode root)
    {
        Stack<TreeNode> stack = new Stack<>();
        List<Integer> list = new ArrayList<>();
        TreeNode curNode = root;
        TreeNode pre = null;
        while(curNode != null || !stack.isEmpty()) {
            while (curNode != null) {
                stack.push(curNode);
                curNode = curNode.left;
            }
            //curNode = stack.pop();不能直接出栈,因为后面有可能不访问,即else中的情况
            curNode = stack.peek();
            //注意:两种情况
            //1、curNode.right == null 没有右子树
            //2、curNode.right == pre  右子树的根为上一次访问的结点
            if (curNode.right == null || curNode.right == pre) {
                list.add(curNode.val);
                stack.pop();  //此处出栈
                pre = curNode;
                curNode = null;
            } else {
                curNode = curNode.right;
            }
        }
        return list;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值