GPLT L2-006. 樹的遍歷【遞歸建樹+bfs輸出】

題目：樹的遍歷

思路：

（1）分別用中序和後序利用遞歸建樹；

（2）按層輸出的話用bfs即可。

代碼：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#define ms(x) memset(x,0,sizeof(x))
using namespace std;
const int maxn = 35;
int lch[maxn],rch[maxn],amid[maxn],rear[maxn],n;
int build(int l1,int r1,int l2,int r2){//遞歸建樹 l1,r1代表後序的左右值，l2,r2代表中序的
    if(l1 > r1) return 0;//葉子節點
    int root = rear[r1];//後序的最後一個值爲根
    int p = 0;
    for(int i=l2;i<=r2;i++){if(amid[i] != root)p++;else break;}//從中序中找根的位置
    lch[root] = build(l1,l1+p-1,l2,l2+p-1);//分別將中序和後序的左子樹存放到左子樹數組中
    rch[root] = build(l1+p,r1-1,l2+p+1,r2);//分別將中序和溝洫的右子樹存放在右子樹數組中
    return root;//返回根結點
}
void bfs(int start){//bfs按層輸出
    queue<int>Q;
    Q.push(start);
    while(!Q.empty()){
        int cur = Q.front();Q.pop();//每次將根輸出
        printf("%d",cur);
        if(lch[cur]) Q.push(lch[cur]);//將當前層的左值入隊
        if(rch[cur]) Q.push(rch[cur]);//將當前層的右值入隊
        if(!Q.empty()) printf(" ");
    }
    printf("\n");
}
int main()
{
    while(~scanf("%d",&n)){
        for(int i=0;i<n;i++) scanf("%d",&rear[i]);
        for(int i=0;i<n;i++) scanf("%d",&amid[i]);
        ms(lch);ms(rch);
        build(0,n-1,0,n-1);
        bfs(rear[n-1]);
    }
    return 0;
}