LeetCode OJ平台上Binary Tree Postorder Traversal题目用java堆栈实现

5 篇文章 0 订阅
1 篇文章 0 订阅

原始题目如下,意为后序遍历二叉树,不要用递归。

Given a binary tree, return the postorder traversal of its nodes' values.

For example:
Given binary tree {1,#,2,3},

   1
    \
     2
    /
   3

return [3,2,1].

Note: Recursive solution is trivial, could you do it iteratively?


后序遍历略微复杂,有两种方法:1,使用额外堆栈,记录节点的右孩子是否访问过;2,将“前序遍历”(中、右、左),然后逆序输出即为后序遍历。两种方法源代码如下:

import java.util.ArrayDeque;
import java.util.LinkedList;
import java.util.List;

public class Solution {
	public static List<Integer> postorderTraversal(TreeNode root){
		ArrayDeque<TreeNode> stack = new ArrayDeque<TreeNode>();
		List<Integer> a = new LinkedList<Integer>();
		List<Integer> b = new LinkedList<Integer>();
		TreeNode tmp = null;
		if(root == null){
			return a;
		}
		else{
			stack.add(root);			
			while(!stack.isEmpty()){
				tmp = stack.removeLast();
				a.add(tmp.val);	
				if(tmp.left != null)
					stack.add(tmp.left);
				if(tmp.right != null)
					stack.add(tmp.right); 
			}
			for(int i = a.size()-1 ; i >= 0; i --)
				b.add(a.get(i));
			return b;
		}
	}
}


import java.util.ArrayDeque;
import java.util.LinkedList;
import java.util.List;

public class Solution {
	public static List<Integer> postorderTraversal(TreeNode root){
		ArrayDeque<TreeNode> stack = new ArrayDeque<TreeNode>();
		ArrayDeque<TreeNode> flag = new ArrayDeque<TreeNode>();//Node in this stack, its right child had been visited
		List<Integer> a = new LinkedList<Integer>();
		TreeNode tmp = null;
		if(root == null){
			return a;
		}
		else{
			stack.add(root);
			while(stack.peekLast().left != null){
				stack.add(stack.peekLast().left);
			}				
			while(!stack.isEmpty()){
				if(stack.peekLast().right == null){//
					tmp = stack.removeLast();
					a.add(tmp.val);					
				}
				else if(!flag.isEmpty() && stack.peekLast().equals(flag.peekLast())){
					// its right child had been visited
					tmp = stack.removeLast();
					a.add(tmp.val);	
					flag.removeLast();
				}
				else{//it has right child and the right child had not been visited
					tmp = stack.peekLast();
					stack.add(stack.peekLast().right);
					flag.add(tmp);
					tmp = stack.peekLast();
					while(tmp.left != null){
						stack.add(tmp.left);
						tmp = tmp.left;
					}
				}
			}
			return a;
		}
	}
	
	public static void main(String args[]){
		List<Integer> c = new LinkedList<Integer>();
		TreeNode p = new TreeNode(1);
		TreeNode p2 = new TreeNode(2);
		TreeNode p3 = new TreeNode(3);
		TreeNode p4 = new TreeNode(4);
		TreeNode p5 = new TreeNode(5);
		TreeNode p6 = new TreeNode(6);
		TreeNode p7 = new TreeNode(7);
		
		p.left = p2;
		p.right = p5;
		p2.left = p3;
		p2.right = p4;
		p3.left = null;
		p3.right = null;
		p4.left = null;
		p4.right = null;
		p5.left = null;
		p5.right = null;
		c = postorderTraversal(p);
		for(int i = 0; i < c.size(); i ++ ){
			System.out.println(c.get(i));
		}
	}
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值