1. 程式人生 > >玩轉二叉樹

玩轉二叉樹

esp int struct style ios tro clu 輸出格式 lag

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

輸入格式:

輸入第一行給出一個正整數N(<=30),是二叉樹中結點的個數。第二行給出其中序遍歷序列。第三行給出其前序遍歷序列。數字間以空格分隔。

輸出格式:

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

輸入樣例:

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

輸出樣例:

4 6 1 7 5 3 2

解題思路:層次遍歷用到的是隊列,和原來不一樣的地方是 這裏先往隊列裏面加入右邊的結點再加入左邊的結點。
#include <iostream>
using namespace std;
#include <stack>
#include <queue>
int a[50];
int b[50];
int flag=1;
struct node
{
    int x,y,z,k;
}t;
queue <struct node> qu; 
int f(int inleft,int inright,int preleft,int preright)
{
    int i;
    struct node p,q;
    t.x=inleft;
    t.y
=inright; t.z=preleft; t.k=preright; qu.push(t); while(!qu.empty()) { p=qu.front(); qu.pop(); if(flag) { cout<<b[p.z]; flag=0; } else cout<<" "<<b[p.z]; for(i=p.x;i<=p.y;i++) {
if(a[i]==b[p.z]) { break; } } int lsize=i-p.x; int rsize=p.y-i; if(rsize>0) { q.x=i+1; q.y=p.y; q.z=p.z+1+lsize; q.k=p.k; qu.push(q); } if(lsize>0) { q.x=p.x; q.y=i-1; q.z=p.z+1; q.k=p.z+lsize; qu.push(q); } } } int main() { int n; cin>>n; for(int i=1;i<=n;i++) cin>>a[i]; for(int i=1;i<=n;i++) cin>>b[i]; f(1,n,1,n); cout<<"\n"; }

玩轉二叉樹