LeetCode: Binary Tree Preorder Traversal 非递归版

没想到非递归的先序遍历对我来说也蛮有挑战性的...

基本思想是从非递归后续遍历改进而来。仍然使用栈作为临时存储区，区别是mark只有0和1两种。

mark=0，代表第一次访问该结点，于是访问+左孩子进栈+mark自增1

mark=1，代表第二次访问该结点，于是出栈该结点+右孩子进栈

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
    struct Tuple { //自定义结构
        int mark;
        TreeNode* ptr;
        Tuple(int m, TreeNode* p): mark(m), ptr(p) {}
    };
public:
    vector<int> preorderTraversal(TreeNode *root) {
        vector<int> vec;
        if (root == NULL) return vec;
        else {
            vector<Tuple> stack;
            stack.push_back(Tuple(0, root));
            while (!stack.empty()) {
                Tuple& temp = stack.back(); //使用引用访问栈顶元素
                switch (temp.mark) {
                    case 0: 
                        vec.push_back(temp.ptr->val); //访问当前结点
                        ++(temp.mark); 、、mark自增
                        if (temp.ptr->left != NULL) stack.push_back(Tuple(0, temp.ptr->left)); //左孩子进栈
                        break;
                    case 1:
                        stack.pop_back(); //当前结点出栈
                        if (temp.ptr->right != NULL) stack.push_back(Tuple(0, temp.ptr->right)); //右孩子进栈
                        break;
                }
            }
            return vec;
        }
        
    }
};