1020. Tree Traversals (25)

/*-----------------------------------------------------------------------
本题采用先根据后序和中序遍历的结果建树，然后再根据建好的树进行层序遍历的方法来完成
-------------------------------------------------------------------------*/
#include<stdio.h>
#include<stdlib.h>
typedef struct queue* q;
typedef struct Node* node;
struct Node{
	int digit;
	node left;
	node right;
};//树节点的基本结构

struct queue{
	node nodee;
	struct queue* next;
};//树的层序遍历要用到队列

q head=NULL,tail=NULL;

node build_tree(int post[],int in[],int amount);//通过后序遍历和中序遍历的结果建树
int find_root(int post[],int in[],int amount);//找到树（或子树）的根节点
void level_order(node root);//对已经建好的树进行层序遍历
void enqueue(node this_node);
node dequeue();

int main(){
	node root;
	root = (node)malloc(sizeof(struct Node));
	int i,num;
	int mid;
	int postorder[30],inorder[30];
	scanf("%d",&num);
	for(i=0;i<num;i++)
		scanf("%d",&postorder[i]);
	for(i=0;i<num;i++)
		scanf("%d",&inorder[i]);

	mid = find_root(postorder,inorder,num);
	root->digit = postorder[num-1];// or   root->digit = inorder[mid];
	root->left = build_tree(postorder,inorder,mid);
	root->right = build_tree(&postorder[mid],&inorder[mid+1],num-mid-1);
	/*--------------------------------------------------------------------------------------------------------
				后序遍历的结果为 左子树+右子树+根节点；中序遍历的结果为 左子树+根节点+右子树
			因此，树的根节点必然是后序遍历的最后一个元素。在中序遍历的结果中找到这个元素，以此为界，左右分别是根的左右子树
			采用递归的方式逐步劈分左右子树，直到树被建完
	---------------------------------------------------------------------------------------------------------*/
	level_order(root);

	return 0;
}

node build_tree(int post[],int in[],int amount){
	node root;
	int mid;
	if(amount == 0)
		return NULL;
	root = (node)malloc(sizeof(struct Node));
	mid = find_root(post,in,amount);
	root->digit = post[amount-1];
	root->left = build_tree(post,in,mid);
	root->right = build_tree(&post[mid],&in[mid+1],amount-mid-1);
	return root;
}
int find_root(int post[],int in[],int amount){
	int i;
	for(i=0;i<amount;i++)
		if(post[amount-1] == in[i])
			return i;
}

/*-----------------------------------------------------------------------
	层序遍历的基本方法（使用队列的数据结构，非递归）好像也没有递归方法=_=||
	1、根入队
	2、根的所有儿子入队
	3、进行一次出队，对出队的元素进行visit，并令出队元素为新的根
	4、从2开始重复步骤，直到队列为空
-------------------------------------------------------------------------*/
void level_order(node root){
	enqueue(root);
	while(root){
		root = dequeue();
		if(root){//not a empty queue
			enqueue(root->left);
			enqueue(root->right);
			printf("%d",root->digit);
			if(head)
				printf(" ");
		}
	}
	return ;
}

/*入队操作*/
void enqueue(node this_node){
	if(this_node==NULL)
		return ;
	if(head == NULL){
		tail =  (q)malloc(sizeof(struct queue));
		head = tail;
	}
	else{
		tail->next = (q)malloc(sizeof(struct queue));
		tail = tail->next;
	}
	tail->nodee = this_node;
	tail->next = NULL;
	return ;
}

/*出队操作*/
node dequeue(){
	if(head == NULL)
		return NULL;
	else {
		node p;
		q temp;
		p = head->nodee;
		temp = head;
		head=head->next;
		free(temp);
		return p;
	}
}