中国大学MOOC数据结构基础习题集、033、Tree Traversals Again

题目链接：http://www.patest.cn/contests/mooc-ds/03-3

题目分析：借助“栈”进行树的后续遍历。栈工作记录中必须注明刚才是在左子树还是在右子树中。

　　每次PUSH，times = 1；

　　每次POP检查栈顶记录的times：如果是1，弹出来变成2压回栈；

　　　　　　　　　　　　　　　　  如果是2，则弹出，放入存放结果的vector中，重复这一过程，直到栈顶times为1。

　　所有PUSH与POP操作执行完毕时，输出vector内的数据和stack中的数据即可。注意要处理最后的空格。

代码分析：

　　头文件声明：1~4　　

　　定义数据结构：5~14

　　按照题目分析所说：15~57

　　输出结果：58~70

#include <iostream>
#include <stack>
#include <vector>
using namespace std;
typedef struct node
{
    int data;
    int times;
    node(int d, int t)
        :data(d), times(t)
    {

    };
} Node;
int main()
{
    int n;
    cin >> n;
    string cmd;
    int x;
    stack<Node> sta;
    vector<int> vec;
    for(int i=0; i<2*n; i++)
    {
        cin >> cmd;
        if(cmd == "Push")
        {
            cin >> x;
            sta.push(Node(x, 1));
        }
        if(cmd == "Pop")
        {
            Node node = sta.top();
            sta.pop();
            if(node.times == 1)
            {
                node.times = 2;
                sta.push(node);
            }
            else if(node.times == 2)
            {
                vec.push_back(node.data);
                while(sta.top().times == 2)
                {
                    vec.push_back(sta.top().data);
                    sta.pop();
                }
                if (sta.size() != 0)
                {
                    node = sta.top();
                    node.times = 2;
                    sta.pop();
                    sta.push(node);
                }
            }
        }
    }
    for(int i=0; i<vec.size(); i++)
    {
        cout << vec[i]<< " ";
    }
    while(sta.size() != 0)
    {
        cout << sta.top().data;
        sta.pop();
        if(sta.size() != 0)
            cout << " ";
    }
    return 0;
}

AC成果：