C4top-玩轉二叉樹（根據前序中序映象反轉後層次遍歷）

kewlgrl發表於2017-03-18
二叉樹
玩轉二叉樹  

給定一棵二叉樹的中序遍歷和前序遍歷，請你先將樹做個鏡面反轉，再輸出反轉後的層序遍歷的序列。所謂鏡面反轉，是指將所有非葉結點的左右孩子對換。這裡假設鍵值都是互不相等的正整數。

輸入格式：

輸入第一行給出一個正整數N（\le≤30），是二叉樹中結點的個數。第二行給出其中序遍歷序列。第三行給出其前序遍歷序列。數字間以空格分隔。

輸出格式：

在一行中輸出該樹反轉後的層序遍歷的序列。數字間以1個空格分隔，行首尾不得有多餘空格。

輸入樣例：

7
1 2 3 4 5 6 7
4 1 3 2 6 5 7

輸出樣例：

4 6 1 7 5 3 2
 
  • 時間限制：400ms
  • 記憶體限制：64MB
  • 程式碼長度限制：16kB
  • 判題程式：系統預設
  • 作者：陳越
  • 單位：浙江大學

題目判定

解題程式
#include<cstdio>
#include<cstring>
#include<cmath>
#include<iomanip>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
#define MAXN 40
#define INF 0x3f3f3f3f
typedef struct node//二叉樹
{
    int data;
    struct node * lchild;
    struct node * rchild;
} btnode;
int n;
int pre[MAXN],in[MAXN];

btnode *creat(int * pre, int * in,int n)//前序，中序，節點數
{
    btnode * b;
    int * p;
    int k;
    if(n<=0) return NULL;
    b=(btnode*)malloc(sizeof(btnode));
    b->data=*pre;
    for(p=in; p<in+n; ++p)
        if(*p==*pre)
            break;
    k=p-in;
    b->lchild=creat(pre+1,in,k);
    b->rchild=creat(pre+k+1,p+1,n-k-1);
    return b;
}
int dis[MAXN],cnt=0;//儲存遍歷到的元素
void level(btnode * b,int n)//層次遍歷
{
    btnode *p;
    btnode *q[n];
    int front,rear;
    front=rear=-1;
    ++rear;
    q[rear]=b;
    while(front!=rear)
    {
        front=(front+1)%n;
        p=q[front];
        dis[cnt++]=p->data;//遍歷時的當前陣列
        //先輸出右孩子再輸出左孩子來實現映象反轉
        if(p->rchild!=NULL)
        {
            rear=(rear+1)%n;
            q[rear]=p->rchild;
        }
        if(p->lchild!=NULL)
        {
            rear=(rear+1)%n;
            q[rear]=p->lchild;
        }
    }
}
int main()
{
#ifdef ONLINE_JUDGE
#else
    freopen("G:/cbx/read.txt","r",stdin);
//freopen("G:/cbx/out.txt","w",stdout);
#endif
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>n;
    for(int i=0; i<n; ++i)
        cin>>in[i];//中序
    for(int i=0; i<n; ++i)
        cin>>pre[i];//前序
    btnode * b;
    b=creat(pre,in,n);//建樹
    level(b,n);//層次遍歷
    for(int i=0; i<cnt-1; ++i)
        cout<<dis[i]<<" ";
    cout<<dis[cnt-1]<<endl;
    return 0;
}


相關文章