1. 程式人生 > 其它 >L2-011 玩轉二叉樹

L2-011 玩轉二叉樹

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;
}