第五章 树 6 AcWing 1576. 再次树遍历

第五章 树 6 AcWing 1576. 再次树遍历

原题链接

AcWing 1576. 再次树遍历

算法标签

树的遍历

思路

#pragma GCC optimize(2)
#pragma GCC optimize(3)
#include<bits/stdc++.h>
#define int long long
#define x first
#define y second
#define ump unordered_map
#define pq priority_queue
#define rep(i, a, b) for(int i=a;i<b;++i)
#define Rep(i, a, b) for(int i=a;i>=b;--i)
#define rep1(i, a, b) for(i=a;i<b;++i)
#define Rep1(i, a, b) for(i=a;i>=b;--i)
using namespace std;
typedef pair<int, int> PII;
const int N = 35;
//int t, n, m, cnt, ans; 
int l[N], r[N];
int n;
// ump<int, int> l, r, pos;
inline int rd(){
   int s=0,w=1;
   char ch=getchar();
   while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
   while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
   return s*w;
}
void put(int x) {
    if(x<0) putchar('-'),x=-x;
    if(x>=10) put(x/10);
    putchar(x%10^48);
}
// 后序递归输出
void dfs(int u, int root){
	// 树为空 
    if(!u){
        return;
    }
    // 递归处理当前节点左右子树
    dfs(l[u], root);
    dfs(r[u], root);
    // 输出根节点
    cout<<u;
    // pat要求行末不允许存在空格 后序遍历最后输出即为根节点 当输出至最后时 不输出空格 
    if(u!=root){
        cout<<" ";
    }
} 
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	stack<int> sta;
	string o;
	int v, type, root, pre=0;
	cin>>n;
	rep(i, 0, 2*n){
	    cin>>o;
	    if(o=="Push"){
	        cin>>v;
	        // 第一次push的值为根
	        if(!pre){
	            root=v;
	        }else{
	        	// 若上次操作为push 则当前节点为上一节点左儿子
	            if(type==0){
	                l[pre]=v;
	            }else{
	            // 若上次操作为pop 则当前节点为上一节点右儿子
	                r[pre]=v;
	            }
	        }
	        // 当前节点入栈
	        sta.push(v);
	        pre=v;
	        type=0;
	    }else{
	    	// 获取当前头节点
	        pre=sta.top();
	        // 当前头节点出栈
	        sta.pop();
	        type=1;
	    }
	}
	dfs(root, root);
	return 0;
}

原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈