L2-006 樹的遍歷
分數 25
給定一棵二叉樹的后序遍歷和中序遍歷，請你輸出其層序遍歷的序列。這里假設鍵值都是互不相等的正整數。
輸入格式：
輸入第一行給出一個正整數N（≤30），是二叉樹中結點的個數。第二行給出其后序遍歷序列。第三行給出其中序遍歷序列。數字間以空格分隔。
輸出格式：
在一行中輸出該樹的層序遍歷的序列。數字間以1個空格分隔，行首尾不得有多余空格。
輸入樣例：
7
2 3 1 5 7 6 4
1 2 3 4 5 6 7
輸出樣例：
4 1 6 3 5 7 2題解
?根據中序遍歷和后序遍歷建立二叉樹，用結構體儲存，然后用bfs輸出層序遍歷。
#include<bits/stdc++.h>
using namespace std;
int n;
int mid[35];
int after[35];
struct node{int l,r;//左右子樹
}t[35];
vector<int> ans;
int build(int l1,int l2,int r1,int r2)
{if(l1>l2) return 0;//樹為空int root=after[r2];//標記根節點int p=l1;//尋找在中序遍歷中的根節點while(mid[p]!=root){p++;}int cnt=p-l1;//左右子樹分別遞歸t[root].l=build(l1,p-1,r1,r1+cnt-1);t[root].r=build(p+1,l2,r1+cnt,r2-1);return root;
}
//bfs輸出層序遍歷
void bfs(int s)
{queue<int> q;q.push(s);while(!q.empty()){int u=q.front();q.pop();ans.push_back(u);if(t[u].l!=0){q.push(t[u].l);}if(t[u].r!=0){q.push(t[u].r);}}
}
int main()
{cin>>n;for(int i=1;i<=n;i++){cin>>after[i];}for(int i=1;i<=n;i++){cin>>mid[i];}//int root=build(mid[1],mid[n],after[1],after[n]);int root=after[n];build(1,n,1,n);bfs(root);int cnt=0;//輸出的時候注意最后一個不能有空格for(auto k:ans){cnt++;if(cnt!=ans.size()){cout<<k<<" ";}else{cout<<k<<endl;}}return 0;
}
