问题描述:
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?
原问题链接:https://leetcode.com/problems/binary-tree-postorder-traversal/
问题分析
和之前问题的讨论类似,关于二叉树的后序遍历,也是有固定的套路可循的。只是二叉树后续遍历的非递归解法确实比较复杂。需要仔细分析。
递归解法
这种解法还是固定的套路,首先递归的去看节点的左子节点,再去看节点的右子节点,再访问当前节点。在访问节点的时候将节点的值加入到列表中。
详细的代码实现如下:
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ public class Solution { public List<Integer> postorderTraversal(TreeNode root) { List<Integer> result = new LinkedList<>(); if(root == null) return result; postorderTraversal(root, result); return result; } private void postorderTraversal(TreeNode node, List<Integer> list) { if(node == null) return; if(node.left != null) postorderTraversal(node.left, list); if(node.right != null) postorderTraversal(node.right, list); list.add(node.val); } }
非递归解法
这种解法思路如下:对于任意一个节点,我们需要访问了它的左右子节点之后才能访问它。所以,我们可以这样来考虑,对于一个节点,先将它入栈,如果它的左右子节点为空,则可以直接访问它。另外,如果它的左右子节点都被访问过了,也可以访问它。如果不是以上的这两种情况,则先后将它的右子节点和左子节点入栈。这样保证了每次先访问左子节点,再访问右子节点。
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ public class Solution { public List<Integer> postorderTraversal(TreeNode root) { List<Integer> result = new LinkedList<>(); LinkedList<TreeNode> stack = new LinkedList<>(); TreeNode cur, pre = null; if(root != null) { stack.push(root); while(!stack.isEmpty()) { cur = stack.peek(); if((cur.left == null && cur.right == null) || (pre != null && (pre == cur.left || pre == cur.right))) { result.add(cur.val); stack.pop(); pre = cur; } else { if(cur.right != null) stack.push(cur.right); if(cur.left != null) stack.push(cur.left); } } } return result; } }