PTA 完全二叉树的层序遍历

题目

一个二叉树，如果每一个层的结点数都达到最大值，则这个二叉树就是完美二叉树。对于深度为 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

题目分析

经分析，完全二叉树的层序遍历中，若以顺序存储的方式逐个存储各个节点的信息，在节点数目范围内，一个节点的左孩子是当前节点所在位置$i\times 2$的位置，右孩子在$i\times 2+1$处。（节点编号从1开始）那么将数组逐个输出则是该完全二叉树的层序遍历结果。【可以进行证明】
由此可知，若将后序遍历的结果按照一定的方法放入完全二叉树该结点在顺序存储中的对应位置，而后将数组从头至尾逐个输出即为层序遍历结果。

代码

#include<iostream>
using namespace std;
typedef struct DataPoint* DP;
struct DataPoint{
	DP left,right;
	int data;
};

struct DataPoint tree[35];
int n;
void init(){//初始化，构建完全二叉树的连接关系
	for(int i = 1;i<=n;++i){
		if(i*2<=n) tree[i].left = &tree[i*2];//左孩子
		else tree[i].left=nullptr;
		if((i*2+1)<=n) tree[i].right = &tree[i*2+1];//右孩子
		else tree[i].right=nullptr;
	}
}

void traverse(struct DataPoint* BT){//后序遍历输入
	if(BT->left!=nullptr) traverse(BT->left);
	if(BT->right!=nullptr) traverse(BT->right);
	scanf("%d",&BT->data);
}

int main(){
	cin >> n;
	init();
	traverse(&tree[1]);
	for(int i = 1;i<n;++i){//逐个输出，结果为层序遍历结果
		cout << tree[i].data<<" ";
	}
	cout <<tree[n].data<<endl;
	return 0;
}

运行结果