[二叉树] 前、中、后序遍历(递归与非递归两种方法)

  • 前序遍历:
    1.根节点为空,直接返回
    2.用cur标记当前结点
    3.cur或栈不为空时:若当前结点不为空,打印,并进栈,一直走到最左边
    3.cur或栈不为空时:若当前结点为空,栈顶元素出栈,让cur指向其右孩子
	//前序遍历:根--左--右
	public void preorderTraversal(TreeNode root) {
        if(root == null)
            return;
        System.out.print(root.val);
        preorderTraversal(root.left);
        preorderTraversal(root.right);
	}
	
	//(非递归方法)
	public void preorderTraversal2(TreeNode root) {
		if(root == null)
			return;
		Stack<TreeNode> stack = new Stack<TreeNode>();	
		//使用cur遍历二叉树的各节点
		TreeNode cur = root;
		while(cur != null || !stack.isEmpty()){
			while(cur != null){
				//当前结点不为空,打印,并进栈,一直走到最左边
				System.out.print(cur.val);
				stack.push(cur);
				cur = cur.left;
			}
			//当前结点为空,栈顶元素出栈,让cur指向其右孩子
			TreeNode top = stack.pop();
			cur = top.right;  
			//若当前结点的右孩子为空,就要看栈是否为空,不为空继续循环
		}
	}
  • 中序遍历:
    1.根节点为空,直接返回
    2.用cur标记当前结点
    3.cur或栈不为空时:若当前结点不为空,直接进栈,一直走到最左边
    3.cur或栈不为空时:若当前结点为空,栈顶元素出栈,并打印,然后让cur指向其右孩子
	//中序遍历
    public void inorderTraversal(TreeNode root) {
    	if(root == null)
    		return;
    	inorderTraversal(root.left);
    	System.out.print(root.val);
    	inorderTraversal(root.right);
    }
    
	//中序遍历(非递归算法)
    public void inorderTraversal2(TreeNode root) {
    	if(root == null)
    		return;
    	TreeNode cur = root;
    	Stack<TreeNode> stack = new Stack<TreeNode>();
    	while(cur != null || !stack.isEmpty()){
    		//当cur不为空时,进栈,一直走到最左端结点
    		while(cur != null){
    			stack.push(cur);
    			cur = cur.left;
    		}
    		//当cur为空时,取栈顶元素并打印,然后让cur指向当前元素的右孩子
    		TreeNode top = stack.pop();
    		System.out.print(top.val);
    		cur = top.right;
    	}
    }
  • 后序遍历:
    1.根节点为空,直接返回
    2.用cur标记当前结点,用prev标记已访问过的结点
    3.cur或栈不为空时:若当前结点不为空,直接进栈,一直走到最左边
    3.cur或栈不为空时:若当前结点为空,取栈顶元素,若该元素的右孩子为空或已被访问,则栈顶元素出栈并打印,同时更新prev和cur
    4.cur或栈不为空时:若该元素的右孩子不为空或没有被访问,则让cur指向其右孩子
    //后序遍历
    public void postorderTraversal(TreeNode root) {
    	if(root == null)
    		return;
    	postorderTraversal(root.left);
    	postorderTraversal(root.right);
    	System.out.print(root.val);
    }
    
    //后序遍历(非递归算法)
    public void postorderTraversal2(TreeNode root) {
    	if(root == null)
    		return;
    	TreeNode cur = root;
    	TreeNode prev = null;
    	Stack<TreeNode> stack = new Stack<TreeNode>();
    	while(cur != null || !stack.isEmpty()){
    		//当cur不为空时,进栈,一直走到最左端结点
    		while(cur != null){
    			stack.push(cur);
    			cur = cur.left;
    		}
    		cur = stack.peek();
    		//若当前结点的右孩子为空或者是当前结点已经访问过,就访问当前结点
    		if(cur.right == null || cur.right==prev){
    			System.out.print(cur.val);
    			stack.pop();
    			prev = cur;
    			cur = null;
    		}
    		//当前结点的右孩子不为空,让当前结点指向有孩子
    		else{
    			cur = cur.right;
    		}
    	}
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值