LeetCode OJ

二叉树的三种遍历，我之前写过这个的迭代实现，详见 这里。

下面是Leetcode这道题的AC代码，同样采用迭代方法，但是我用了一个ArrayList去记录已经出栈的节点，这样当我考察栈顶节点是否直接弹出时，只要看它是否有左右子节点或者是它的左右子节点是否已经被弹出。这种方法可能在时间复杂度上更高。

/**
     * 后序遍历
     * @param root
     * @return
     */
    public ArrayList<Integer> postOrderTraversal(TreeNode root)
    {
        ArrayList<Integer> r = new ArrayList<Integer>();
        if(root == null)
            return null;
        LinkedList<TreeNode> stack = new LinkedList<TreeNode>();
        
        ArrayList<TreeNode> tr = new ArrayList<TreeNode>();
        stack.push(root);
        while(!stack.isEmpty())
        {
            TreeNode temp = stack.peek();
            if((temp.left == null && temp.right == null) ||
                    tr.contains(temp.left) || tr.contains(temp.right))
            {
                stack.pop();
                r.add(temp.val);
                tr.add(temp);
            }else {
                if(temp.right!=null)
                    stack.push(temp.right);
                if(temp.left!=null)
                    stack.push(temp.left);
            }
        }
        return r;
    }
    /**
     * 先序遍历
     * @param root
     * @return
     */
    public ArrayList<Integer> preorderTraversal(TreeNode root)
    {
        if(root == null)
            return null;
        LinkedList<TreeNode> stack = new LinkedList<TreeNode>();
        ArrayList<Integer> r = new ArrayList<Integer>();
        stack.push(root);
        while(!stack.isEmpty()){
            TreeNode temp = stack.pop();
            r.add(temp.val);
            if(temp.right!=null)
                stack.push(temp.right);
            if(temp.left!=null)
                stack.push(temp.left);
        }
        return r;
    }