【每日刷题】二叉树的前序遍历

题目地址

https://leetcode-cn.com/problems/binary-tree-preorder-traversal/

题目描述：二叉树的前序遍历

给定一个二叉树，返回它的 前序 遍历。

示例:

输入: [1,null,2,3]

  1
   \
    2
   /
  3 

输出: [1,2,3]

解答

方法一：递归。

太简单了，不再赘述。

方法二：迭代

用栈作为辅助空间，将当前结点压栈，设置一个 count, 当 count == 1 时将左孩子压栈，当 count == 2 时将右孩子压栈。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        if( !root)
            return {};
        
        stack< pair<TreeNode*, int>> record;
        record.push( {root, 0});
        
        vector<int> result;
        while( !record.empty()){
            if( record.top().second == 0)
                result.push_back( record.top().first->val), record.top().second++;
            else if( record.top().second == 1){
                record.top().second++;
                if( record.top().first->left)
                    record.push( {record.top().first->left, 0});
            }       
            else if( record.top().second == 2){
                TreeNode* temp = record.top().first;
                record.pop();
                if( temp->right)
                    record.push( {temp->right, 0});
            }
        }
        return result;
    }
};