L2-011 玩轉二叉樹 (25 分)(建樹+後序翻轉)
阿新 • • 發佈:2021-02-05
給定一棵二叉樹的中序遍歷和前序遍歷,請你先將樹做個鏡面反轉,再輸出反轉後的層序遍歷的序列。所謂鏡面反轉,是指將所有非葉結點的左右孩子對換。這裡假設鍵值都是互不相等的正整數。
輸入格式:
輸入第一行給出一個正整數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<bits/stdc++.h> using namespace std; typedef struct tnode { int data; struct tnode*left; struct tnode*right; }Tnode,*Bitree; Bitree Rebitree(int pre[],int p,int innode[],int q,int n); void reverse(Bitree T); void cengxu(Bitree T); int main() { int n; cin>>n; int innode[n]; int pre[n]; int i; for(i=0;i<n;i++){ cin>>innode[i]; } for(i=0;i<n;i++){ cin>>pre[i]; } Bitree T; T=Rebitree(pre,0,innode,0,n); reverse(T); cengxu(T); return 0; } Bitree Rebitree(int pre[],int p,int innode[],int q,int n) { if(n<=0){ return NULL; } Bitree T; T=(Bitree)malloc(sizeof(Tnode)); T->data=pre[p]; int i; for(i=q;i<q+n;i++){ if(innode[i]==pre[p]){ break; } } if(i==q){ T->left=NULL; }else{ T->left=Rebitree(pre,p+1,innode,q,i-q); } if(i==q+n){ T->right=NULL; }else{ T->right=Rebitree(pre,p+i-q+1,innode,i+1,q+n-i-1); } return T; } void reverse(Bitree T) { if(T==NULL){ return; } reverse(T->left); reverse(T->right); if(T->left||T->right){ Bitree temp; temp=T->left; T->left=T->right; T->right=temp; } } void cengxu(Bitree T) { if(T==NULL){ return ; } queue<Bitree> v; v.push(T); int flag=0; while(!v.empty()) { Bitree temp; temp=v.front(); if(temp->left){ v.push(temp->left); } if(temp->right){ v.push(temp->right); } if(flag==0){ cout<<temp->data; flag=1; }else{ cout<<" "<<temp->data; } v.pop(); } }