PAT_甲级_1086 Tree Traversals Again (25point(s)) (C++)【树的遍历/先序+中序=后序】-CSDN博客

本文链接：https://blog.csdn.net/qq_41528502/article/details/104767478

1，题目描述

implemented：使生效; 贯彻; 执行; 实施；

Sample Input:

6
Push 1
Push 2
Push 3
Pop
Pop
Push 4
Pop
Pop
Push 5
Push 6
Pop
Pop

Sample Output:

3 4 2 6 5 1

题目大意

用栈的一些操作，模拟树的构建，并输出树的后序遍历。

2，思路

将入栈的序列，看作先序遍历；出栈的序列看作中序遍历；（根据例子中的图形得出的结论）
根据树的遍历特点（先序遍历第一个节点为根节点，找到此节点在中序遍历中的位置，即可将树划分为左右两子树，具体可参考@&再见萤火虫&【PAT_甲级_1020 Tree Traversals (25分) (C++)【树的遍历】】）将根节点依次放入postOrder中（由于是将根节点逐个存入，故为了保证正确顺序，先遍历右子树，再遍历左子树，最后逆序输出）；

对函数的补充说明：

限定中序遍历的首尾位置head与tail，和从首到尾的节点代表子树的根节点root；
有这些变量，就可以根据根节点，将树递归地划分下去：root确定根节点在子树中的位置，根据head和tail可以计算左右子树的节点数目，从而继续向下遍历

3，AC代码

#include<bits/stdc++.h>
using namespace std;
int N;
vector<int> inOrder, preOrder, postOrder;
void getPostOrder(int head, int tail, int root){ //head与tail针对中序遍历 root针对先序遍历
    if(head > tail) return;
    postOrder.push_back(preOrder[root]);
    for(int i = head; i <= tail; i++){
        if(inOrder[i] == preOrder[root]){
            getPostOrder(i + 1, tail, root + i - head + 1);//先遍历右子树
            getPostOrder(head, i - 1, root + 1);
            return;
        }
    }
}
int main(){
#ifdef ONLINE_JUDGE
#else
    freopen("1.txt", "r", stdin);
#endif // ONLINE_JUDGE

    stack<int> sta;
    string s;
    cin>>N;
    int num;
    for(int i = 0; i < 2 * N; i++){
        cin>>s;
        if(s[1] == 'u'){
            cin>>num;
            sta.push(num);
            preOrder.push_back(num);    //push送入先序遍历
        }else{
            num = sta.top();
            sta.pop();
            inOrder.push_back(num);     //pop送入中序遍历
        }
    }
    getPostOrder(0, N - 1, 0);
    cout<<postOrder[N - 1];             //得到的是后序遍历的逆序 所以倒着输出
    for(int i = N - 2; i >= 0; i--)
        cout<<' '<<postOrder[i];
    return 0;
}