考察二叉树的遍历
题目大意:
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;
}