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;
}

}
};