[PAT-A 1020]Tree Traversals

在这里插入图片描述
考察二叉树的遍历
题目大意:
1.给出一颗二叉树的后序遍历序列和中序遍历序列,求这颗二叉树的层序遍历序列

思路:
记后序序列区间[postL,postR],中序序列区间[inL,inR]。
后根序列中的最后一个元素是整个二叉树的根节点,在中序序列中寻找in[k]==post[postR]
则左子树中序序列区间[inL,k-1],左子树个数numLeft=k-inL,则左子树后序序列区间[postL,postL+numLeft-1].
右子树中序序列区间[k+1,inR],右子树后序序列区间[post+numLeft,postR-1];

如此递归建立树,再进行层序遍历即可,递归边界:后序序列长度小于0,即postL>postR

AC代码:

//PAT_A 1020
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn = 50;
struct  node{
	int data;
	node* lchild;
	node* rchild;
};
int pre[maxn], in[maxn], post[maxn];//先序,中序,后序
int n;//节点个数
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 = inL;
	for (k = inL; k <= inR; k++) {
		if (in[k] == root->data)break;//中序中找这个根节点元素,将根节点一分为2
	}
	int numLeft = k - inL;//左子树节点个数
	root->lchild = create(postL, postL + numLeft - 1, inL, k - 1);//递归建立左右子树
	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() {
	(void)scanf("%d", &n);
	for (int i = 0; i < n; i++) {
		(void)scanf("%d", &post[i]);
	}
	for (int i = 0; i < n; i++) {
		(void)scanf("%d", &in[i]);
	}
	node* root = create(0, n - 1, 0, n - 1);
	bfs(root);
	return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值