【1020】Tree Traversals (25 分)

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<string.h>
#include<algorithm>  
#include<map>
#include<vector>
#include<queue> 
using namespace std;  
const int maxn=50;
struct node{
  int data;
  node* lchild;
  node* rchild;
};
int pre[maxn],in[maxn],post[maxn];//先序、中序及后序
int n;//结点个数
//当前二叉树的后序序列区间为[postL,postR],中序序列区间为[inL,inR]
//create函数返回构建出的二叉树的根结点地址
node* create(int postL,int postR,int inL,int inR){
  if(postL>postR){
    return NULL;
  }
  node* root=new node;//新建一个新的结点，用来存放当前二叉树的根结点
  root->data=post[postR];//新结点的数据域为根结点的值
  int k;//这个k不要在for里面定义了，注意有效域
  for(k=inL;k<inR;k++){
    if(in[k]==post[postR]){  //在中序序列中找到in[k]==pre[L]的结点
      break;
    }
  }
  int numLeft=k-inL; //左子树的结点个数
  //返回左子树的根结点地址，赋值给root的左指针
  root->lchild=create(postL,postL+numLeft-1,inL,k-1);
  //返回右子树的根结点地址，赋值给root的右指针
  root->rchild=create(postL+numLeft,postR-1,k+1,inR);
  return root;  //返回根结点地址
}
int num=0; //已经输出的结点个数
void BFS(node* root){
  queue<node*> q; //注意队列里是存地址!!!!!!
  q.push(root);  //将根结点地址入队
  while(!q.empty()){
    node* now=q.front();  //取出队首元素
    q.pop();
    printf("%d",now->data); //访问队首元素
    num++;
    if(num<n) printf(" ");
    if(now->lchild != NULL) q.push(now->lchild); //左子树非空
    if(now->rchild != NULL) q.push(now->rchild); //右子树非空
  }
}
int main(){   
  scanf("%d",&n);
  for(int i=0;i<n;i++){
    scanf("%d",&post[i]);
  }
  for(int i=0;i<n;i++){
    scanf("%d",&in[i]);
  }
  node* root=create(0,n-1,0,n-1);  //建树
  BFS(root);
  system("pause");
    return 0;   
}