文章目錄

• • 題目分析
  • 題目鏈接

題目分析

做過前面幾道題，發(fā)現(xiàn)這道題就是一道模板題，遞歸建樹即可。
還是使用筆者熟悉的hash表來找根，進(jìn)行優(yōu)化。 請移步至筆者的另一篇文章：PAT甲級1020 Tree Traversals：[C++題解]樹的遍歷、由中序序列和后序序列遞歸建樹

ac代碼
寫復(fù)雜了，下文有簡單版。

#include<bits/stdc++.h> using namespace std;const int N = 5e4 +10;unordered_map<int,int> l,r,pos; int n ,pre[N],in[N]; vector<int> post;//前序遍歷和中序遍歷建樹 int build(int il ,int ir ,int pl ,int pr){int root = pre[pl];int k =pos[root];if(il< k) l[root] =build(il ,k-1, pl+1, pl+1+k-1-il);if(k< ir) r[root] = build(k+1,ir, pl+k-il+1, pr);return root; }//后序遍歷 void dfs(int u){if(u==0) return;dfs(l[u]);dfs(r[u]);post.push_back(u); }int main(){cin>> n;for(int i =0; i< n;i ++ ) cin>> pre[i];for(int i=0; i< n ;i++){cin>>in[i];pos[in[i]] = i;}int root = build(0 ,n-1, 0 ,n-1);dfs(root);cout<<post[0]; }

ac代碼：簡潔版

簡潔版代碼：畢竟僅僅需要輸出后序遍歷的第一個結(jié)點。
直接在建樹的過程中賦值第一個后序遍歷的點即可，不用再寫dfs后序遍歷。

#include<bits/stdc++.h> using namespace std;const int N = 5e4 +10;unordered_map<int,int> pos; int n ,pre[N],in[N]; int post; //后序遍歷第一個結(jié)點void build(int il ,int ir ,int pl ,int pr){int root = pre[pl];int k =pos[root];if(il< k) build(il ,k-1, pl+1, pl+1+k-1-il);if(k< ir) build(k+1,ir, pl+k-il+1, pr);if(post==0) post =root; }int main(){cin>> n;for(int i =0; i< n;i ++ ) cin>> pre[i];for(int i=0; i< n ;i++){cin>>in[i];pos[in[i]] = i;}build(0 ,n-1, 0 ,n-1);cout<<post; }

題目鏈接

PAT甲級1138 Postorder Traversal

總結(jié)

以上是生活随笔為你收集整理的PAT甲级1138 Postorder Traversal：[C++题解]前序遍历和中序遍历建树的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。