重建二叉树

题目1385:重建二叉树  http://ac.jobdu.com/problem.php?pid=1385

时间限制:1 秒内存限制:32 兆特殊判题:否提交:4786解决:1410
题目描述:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并输出它的后序遍历序列。

输入:
输入可能包含多个测试样例,对于每个测试案例,
输入的第一行为一个整数n(1<=n<=1000):代表二叉树的节点个数。
输入的第二行包括n个整数(其中每个元素a的范围为(1<=a<=1000)):代表二叉树的前序遍历序列。
输入的第三行包括n个整数(其中每个元素a的范围为(1<=a<=1000)):代表二叉树的中序遍历序列。
输出:
对应每个测试案例,输出一行:
如果题目中所给的前序和中序遍历序列能构成一棵二叉树,则输出n个整数,代表二叉树的后序遍历序列,每个元素后面都有空格。
如果题目中所给的前序和中序遍历序列不能构成一棵二叉树,则输出”No”。
样例输入:
8
1 2 4 7 3 5 6 8
4 7 2 1 5 3 8 6
8
1 2 4 7 3 5 6 8
4 1 2 7 5 3 8 6
样例输出:
7 4 2 5 8 6 3 1 

No


int Findrootvalue(int *InOrder,int  rootvalue,int TreeLenght){
	for (int i = 0; i < TreeLenght; i++)
	{
		if (InOrder[i] == rootvalue)
			return i;
	}
	return -1;
}
int Rebuild(int *PreOrder, int *InOrder, int TreeLenght, TreeNode **root)
{
	if (TreeLenght == 0)
	{
		
		*root = NULL;
		return 0;
	}
	else{
		int rootvalue = PreOrder[0];
		int tag = Findrootvalue(InOrder, rootvalue, TreeLenght);
		if (tag == -1) return -1;

		*root = (TreeNode*)malloc(sizeof(TreeNode));
		(*root)->intvalue = rootvalue;

		int LeftInOrder[1000], RightInOrder[1000], LeftPreOrder[1000], RightPreOrder[1000];
		
		for (int i = 0; i < tag; i++)
		{
			LeftInOrder[i] = InOrder[i];
			
		}
		
		for (int i = tag + 1; i < TreeLenght; i++)
			RightInOrder[i - tag - 1] = InOrder[i];
		
		for (int i = 1; i <= tag; i++)
		{
			LeftPreOrder[i - 1] = PreOrder[i];
			
		}
		for (int i = tag + 1; i < TreeLenght; i++)
			RightPreOrder[i - tag - 1] = PreOrder[i];
		int jud1, jud2;
		jud1 = Rebuild(LeftPreOrder, LeftInOrder, tag, &((*root)->pLeft));
		if (jud1 == -1) return -1;
		jud2 = Rebuild(RightPreOrder, RightInOrder, TreeLenght - tag-1, &((*root)->pRight));
		if (jud2 == -1) return -1;
	}
	return 0;
}
//后序遍历输出
void PostOrderTraversal(TreeNode *root)
{
	if (root != NULL){
		PostOrderTraversal(root->pLeft);
		PostOrderTraversal(root->pRight);
		printf("%d ", root->intvalue);
	}
	return;
}
int main(){
	//读入信息
	int num;
	int PreOrder[1000 + 10], InOrder[1000 + 10];
	while (~scanf(num)){
		for (int i = 0; i < num; i++)
			scanf(PreOrder[i]);
		for (int i = 0; i < num; i++)
			scanf(InOrder[i]);
		TreeNode * root = NULL;
    //根据前缀中缀建树
		int temp = Rebuild(PreOrder, InOrder, num, &root);
	//判断建树是否成功,成功为0,失败为-1
		if (temp == -1){
			printf("N0");
		}
		else
		if (temp == 0){
			PostOrderTraversal(root);
			printf("\n");
		}

	}
}



你可能感兴趣的:(重建二叉树)