主体部分代码:
需要知道先序、中序序列的区间
//当前先序序列区间为[preL,preR],中序序列区间为[inL,inR],返回根节点地址
node* create(int preL,int preR,int inL,int inR){
if(preL>preR){
return NULL;//先序序列长度小于等于0,直接返回
}
node* root = new node;//新建一个新的节点,用来存放当前二叉树的根节点
root->data = pre[preL];//新节点的数据域为根节点的值
int k;
for(k = inL;k <= inR;k++){
if(in[K] == pre[preL]){//在中序序列中找到in[K] == pre[L]的节点
break;
}
}
int numLeft = k - inL;//左子树的节点个数
//左子树的先序区间为[pre+1,preL+numLeft],中序区间为[inL,k-1]
//返回左子树的根节点地址,赋值给root的左指针
root->lchild = create(preL+1,preL+numLeft,inL,k-1);
//右子树的先序区间为[preL+numLeft+1,preR],中序区间为[k+1,inR]
//返回右子树跟节点地址,赋值给root的右指针
root->rchild = create(preL+numLeft+1,preR,k+1,inR);
return root;//返回根节点地址
}
《算法笔记》P296
#include <stdio.h>
#include <queue>
#include <string.h>
#include <algorithm>
using namespace std;
const int maxn = 50;
struct node{
int data;
node* lchild;
node* rchild;
};
int in[maxn],post[maxn];//两个用来存放中序、后序的数组
int n;//节点个数
node* creat(int postL,int postR,int inL,int inR){
if(postL>postR){
return NULL;//边界
}
node* root = new node;//新建一个新的节点,用来存放当前二叉树的根节点
root->data = post[postR];//新节点的数据域为根节点的值
int k;
for(k = inL;k <= inR;k++){
if(in[k] == post[postR]){//在中序序列中找到in[k] == pre[L]的节点
break;
}
}
int numLeft = k - inL;//左子树的节点个数
//返回左子树的根节点地址,赋值给root的左指针
root->lchild = creat(postL,postL+numLeft-1,inL,k-1);
//返回右子树的根节点地址,赋值给root的右指针
root->rchild = creat(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 = creat(0,n-1,0,n-1);
BFS(root);
return 0;
}