leetcode: Binary Tree Postorder Traversal

问题描述:

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;
    }
} 

猜你喜欢

转载自shmilyaw-hotmail-com.iteye.com/blog/2309940