1. 程式人生 > >中序遍歷和層序遍歷還原二叉樹

中序遍歷和層序遍歷還原二叉樹

遞迴是將大問題分解成小問題求解,遞迴函式中輸入是層序遍歷和中序遍歷,層序遍歷的第一個值一定是根通過根和中序遍歷將樹分成左根右的情況,左子樹在層序遍歷中找到自己對應的層序遍歷進行下一次遞迴,右子樹同理
#include "stdio.h" 
using namespace std;

struct Node
{
	int data;
	Node *leftN;
	Node *rightN;
} node[32];

int cnt = 0;

Node *create()
{
	node[cnt].leftN = NULL;
	node[cnt].rightN = NULL;
	int data = 0;
	return &node[cnt++];
}

void preOrder(Node *n,int num)
{
	if(num==0)
	{
		printf("%d",n->data);
	}
	else
	{
		printf(" %d",n->data);	
	}
	
	if(n->leftN!=NULL)
	{
		preOrder(n->leftN,++num);
	}
	
	if(n->rightN!=NULL)
	{
		preOrder(n->rightN,++num);
	}
}

void lasOrder(Node *n,int num)
{
	
	if(n->leftN!=NULL)
	{
		lasOrder(n->leftN,++num);
	}
	
	if(n->rightN!=NULL)
	{
		lasOrder(n->rightN,++num);
	}
	if(num>=cnt-1)
	{
		printf("%d",n->data);
	}
	else
	{
		printf("%d ",n->data);	
	}
}

Node *insert(int l[],int n,int m[],int p,int lum)
{
	Node *t = create();
	//取出層序遍歷的第一個,並在中序遍歷中查詢它的位置
	int midI = 0;
	for(int i = p;i<=lum;i++) 
	{
		if(m[i]==l[0])
		{
			midI = i;
			break;
		}
	}
	t->data = m[midI];
	if(p==lum)
	{
		return t;
	}
	//先得出左子樹和右子樹的範圍
	int lp=-1,ll=-1,rp=-1,rl=-1;
	if(p<midI) 
	{
		lp = p;
		ll = midI -1;
		//根據這個在層序遍歷中找到層序遍歷 
		int index = 0;
		int at[32] = {0};
		for(int j = 0;j<n;j++) //這裡是層序遍歷 
		{
			for(int i = lp;i<=ll;i++)			
			{
				if(m[i]==l[j])
				{
					at[index++] = m[i];
				}
			}
		}
		//現在at中存在的就是左子樹的層次遍歷
		t->leftN = insert(at,index,m,lp,ll);
	}
	if(midI<lum)
	{
		rp = midI+1;
		rl = lum;
		//根據這個在層序遍歷中找到層序遍歷 
		int index = 0;
		int at[32] = {0};
		for(int j = 0;j<n;j++) 
		{
			for(int i = rp;i<=rl;i++ )			
			{
				if(m[i]==l[j])
				{
					at[index++] = l[j];
				}
			}
		}
		//現在at中存在的就是左子樹的層次遍歷
		t->rightN = insert(at,index,m,rp,rl);
	}
	return t;	
}

int main()
{
	int n = 0;
	scanf("%d",&n);
	int l[32] = {0};
	int m[32] = {0};
	for(int i = 0;i<n;i++)
	{
		scanf("%d",&l[i]);
	}
	for(int i = 0;i<n;i++)
	{
		scanf("%d",&m[i]);
	}
	Node * t = insert(l,n,m,0,n-1);
	preOrder(t,0);
	printf("\n");
	lasOrder(t,0);
	return 0;
}