[leetcode] Binary Tree Preorder Traversal

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

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

   1
    \
     2
    /
   3

 

return [1,2,3].

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

给定一棵二叉树，使用非递归的方法进行先序遍历。

用一个栈来保存回溯的节点，对于每个节点，都先访问栈顶元素，右儿子非空入栈，左儿子非空入栈的顺序进行，直到栈为空。

代码如下：

class Solution {
    public:
        vector<int> preorderTraversal(TreeNode *root) {
            vector<int> res;
            stack<TreeNode *> st;
            TreeNode * p = root;
            st.push(p);
            if( root == NULL )
            {
                return res; 
            }
            while( st.size() != 0 )
            {
                p = st.top();
                st.pop();
                res.push_back(p->val);
                if( p->right != NULL )
                {
                    st.push(p->right);
                }
                if( p->left != NULL )
                {
                    st.push(p->left);
                }
            }
            return res;
        }
};