PAT (Advanced Level) Practice 1086 Tree Traversals Again (25 分)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Nightmare_ak/article/details/84555095

用栈模拟即可

#include<cstdio>
#include<vector>
#include<stack>
using namespace std;

const int N=30+5;

stack<int> stk;
vector<int> ans;
int ch[N][2],key[N];

void dfs(int u)
{
    if(ch[u][0]) dfs(ch[u][0]);
    if(ch[u][1]) dfs(ch[u][1]);
    ans.push_back(key[u]);
}

int main()
{
    int n;scanf("%d",&n);
    int rt=0,cnt=0;
    for(int i=0;i<n*2;i++)
    {
        char op[10];
        scanf("%s",op);
        if(op[1]=='u')
        {
            int x;scanf("%d",&x);
            key[++cnt]=x;
            if(ch[rt][0]) ch[rt][1]=cnt;
            else ch[rt][0]=cnt;
            stk.push(rt=cnt);
        }
        else rt=stk.top(),stk.pop();
    }
    dfs(ch[0][0]);
    for(int i=0;i<(int)ans.size();i++)
        printf("%d%c",ans[i]," \n"[i+1==(int)ans.size()]);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Nightmare_ak/article/details/84555095