[LeetCode 144] Binary Tree Preorder Traversal（迭代法）

题目内容

144 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?
题目来源

题目简述

二叉树前序遍历的迭代方法

题目分析

用栈记录节点访问的顺序。
首先，将根节点入栈，访问该节点。
其次，对于栈中的每个节点，进行以下迭代操作。
若：
1.左子树非空
左子节点入栈
2.左子树为空：
此时分为以下两种情况：
1.栈顶节点的右子树非空
访问栈顶节点的右子节点， 将原栈顶节点出栈，右子节点入栈。
2.栈顶节点的右子树为空
原栈顶节点出栈。
完成上述操作直至栈为空。

代码示例

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
#include<stack>
class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        stack<TreeNode *> track;
        vector<int> res;
        TreeNode *temp=root;
        while(temp!=NULL||!track.empty())
        {
            if(temp)
            {
            while(temp)
            {
                res.push_back(temp->val);
                track.push(temp);
                temp=temp->left;
            }
            }
            else{
                temp=track.top()->right;
                track.pop();

            }   
        }
        return res；   
    }
};