L2-011 玩轉二叉樹
阿新 • • 發佈:2022-03-13
L2-011 玩轉二叉樹 (25 分)
給定一棵二叉樹的中序遍歷和前序遍歷,請你先將樹做個鏡面反轉,再輸出反轉後的層序遍歷的序列。所謂鏡面反轉,是指將所有非葉結點的左右孩子對換。這裡假設鍵值都是互不相等的正整數。
輸入格式:
輸入第一行給出一個正整數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; struct NODE{ int data; NODE* left; NODE* right; }node,*pnode; queue<NODE*> q; NODE* newnode(int data){ NODE* nnode=(NODE*)malloc(sizeof(NODE)); nnode->data=data; nnode->left=NULL; nnode->right=NULL; return nnode; } NODE* build(int* prev,int* in,int n){ NODE* root; if(n==0) return NULL; else if(n==1){ return newnode(*prev); } else{ int i; root=newnode(*prev); for(i=0;i<n;i++){ if(*(in+i)==root->data) break; } root->left=build(prev+1,in,i); root->right=build(prev+1+i,in+i+1,n-i-1); } return root; } int main(){ int n,prevnode[50],innode[50]; NODE* p; cin >> n; for(int i=0;i<n;i++){ cin >> innode[i]; } for(int i=0;i<n;i++){ cin >> prevnode[i]; } int* prevf=prevnode; int* inf=innode; pnode=build(prevf,inf,n); cout << pnode->data; q.push(pnode); while(!q.empty()){ p=q.front(); q.pop(); if(p->right){ cout << " " << p->right->data; q.push(p->right); } if(p->left){ cout << " " << p->left->data; q.push(p->left); } } return 0; }