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

爷,独闯天下 提交于 2020-02-01 18:29:12

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

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



标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!