PTA完全二叉树的层序遍历 (25 分)

完全二叉树的层序遍历 (25 分)

一个二叉树，如果每一个层的结点数都达到最大值，则这个二叉树就是完美二叉树。对于深度为 D 的，有 N 个结点的二叉树，若其结点对应于相同深度完美二叉树的层序遍历的前 N 个结点，这样的树就是完全二叉树。

给定一棵完全二叉树的后序遍历，请你给出这棵树的层序遍历结果。

输入格式：

输入在第一行中给出正整数 N（≤30），即树中结点个数。第二行给出后序遍历序列，为 N 个不超过 100 的正整数。同一行中所有数字都以空格分隔。

输出格式：

在一行中输出该树的层序遍历序列。所有数字都以 1 个空格分隔，行首尾不得有多余空格。

输入样例：

8
91 71 2 34 10 15 55 18

输出样例：

18 34 55 71 2 10 15 91

---

Solution

使用数组存储完全二叉树。核心是build函数。

#include<iostream>
using namespace std;

struct node{
	int val,l=0,r=0;
}tree[32];
void build(int x){//递归建树
	if(tree[x].l)
        build(tree[x].l);
	if(tree[x].r)
        build(tree[x].r);
	cin>>tree[x].val;
}
int main(){
    int n; 
    cin>>n;
    for(int i=1;i<=n;i++){//标记有效节点
    	if(2*i<=n) tree[i].l=2*i;
    	if(2*i+1<=n) tree[i].r=2*i+1;
	}
	build(1);
	//建好后按顺序遍历数组即可
	cout<<tree[1].val;
	for(int i=2;i<=n;i++)
	    cout<<" "<<tree[i].val;
    cout<<endl;
	return 0;
}

同理build可用于前序和中序遍历建树

void build(int x){//前序
	cin>>tree[x].val;
	if(tree[x].l)
        build(tree[x].l);
	if(tree[x].r)
        build(tree[x].r);
}

void build(int x){//中序
	if(tree[x].l)
        build(tree[x].l);
    cin>>tree[x].val;
	if(tree[x].r)
        build(tree[x].r);
}