pat 甲級 1099(二叉樹前序+層序遍歷)
阿新 • • 發佈:2018-11-19
題目連結:https://pintia.cn/problem-sets/994805342720868352/problems/994805367987355648
思路:
(1)建立二叉樹
(2)將b陣列排序得到二叉樹的前序遍歷
(3)層序遍歷二叉樹
注意:二叉樹前序遍歷是先判斷節點是否為空(這裡卡了好久,竟然忘了最基本的前序遍歷,真是太傻比了)。
#include<iostream> #include<cstdio> #include<cstring> #include<queue> #include<algorithm> #include<vector> using namespace std; const int maxn = 220; struct Node{ int l,r,data; }; Node a[maxn]; int b[maxn],cnt,n; vector <int> vc; void it(int pos) { if(pos!=-1) { it(a[pos].l); a[pos].data=b[cnt++]; it(a[pos].r); } } void levelorder() { queue <int> q; q.push(0); while(!q.empty()) { int tp=q.front(); q.pop(); vc.push_back(a[tp].data); if(a[tp].l!=-1) q.push(a[tp].l); if(a[tp].r!=-1) q.push(a[tp].r); } } int main(void) { int i,n,j; cin>>n; for(i=0;i<n;i++) cin>>a[i].l>>a[i].r; for(i=0;i<n;i++) cin>>b[i]; sort(b,b+n); cnt=0; it(0); vc.clear(); levelorder(); //for(i=0;i<n;i++) cout<<a[i].data<<" "; for(i=0;i<vc.size()-1;i++) printf("%d ",vc[i]); printf("%d\n",vc[i]); return 0; }