(算法练习)——【先序、中序遍历确定二叉树】PAT A1020 Tree Traversals

主体部分代码:
需要知道先序、中序序列的区间

//当前先序序列区间为[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;
}



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值