基础题练习——重建二叉树

  
  

   
   
    
    题目描述：
   
   
   
   

    
    
输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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”。
   
   
  
  
 
 

  
  
   
   
    
    样例输入：
   
   
   
   

    
    
81 2 4 7 3 5 6 84 7 2 1 5 3 8 681 2 4 7 3 5 6 84 1 2 7 5 3 8 6

    
    


    
    

     
     
      
      样例输出：
     
     
     
     


      
      
7 4 2 5 8 6 3 1 No

     
     
    
    

    
    


    
    
 
    
    
 
    
    
 
    
    
#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;

struct TreeNode
{
	int val;
	TreeNode *leftChild;
	TreeNode *rightChild;
};

TreeNode* RebuildTree(int *Pre, int* In, int n, bool& valid)
{
	if(!valid)
		return NULL;
	if(n<=0)
		return NULL;
	int rootVal = Pre[0];
	int rootPosofIn = -1;
	for(int i=0; i<n; i++)
		if(In[i]==rootVal)
		{
			if(rootPosofIn==-1)
				rootPosofIn=i;
			else//exist many same index
			{
				valid = false;
				return NULL;
			}
		}
	if(rootPosofIn==-1)//can not locate root
	{
		valid =false;
		return NULL;
	}
	int leftCount = rootPosofIn;
	int rightCount = n-rootPosofIn-1;
	TreeNode *newRoot = new TreeNode();
	newRoot->val = rootVal;
	newRoot->leftChild = RebuildTree(Pre+1,In,leftCount,valid);
	newRoot->rightChild = RebuildTree(Pre+leftCount+1,In+leftCount+1,rightCount,valid);
	return newRoot;
}

int PreSeq[1000];
int InSeq[1000];

void PostOrderTraverse(TreeNode *root)
{
	if(root==NULL)
		return;
	PostOrderTraverse(root->leftChild);
	PostOrderTraverse(root->rightChild);
	cout<<root->val<<" ";

}

int main()
{
	//freopen("in.txt","r",stdin);
	int n;
	while(cin>>n)
	{
		for(int i=0; i<n; i++)
			scanf("%d",PreSeq+i);
		for(int i=0; i<n; i++)
			scanf("%d",InSeq+i);

		bool validTag =true;
		TreeNode* pRoot = RebuildTree(PreSeq,InSeq,n,validTag);
		if(validTag && pRoot!=NULL)
		{
			PostOrderTraverse(pRoot);
			cout<<endl;
		}
		else
			cout<<"No"<<endl;
	}
	return 0;
}