题目大意:给出一个序列,构造是完全二叉树的BST,输出层序遍历结果。
这一题要同时结合完全二叉树和BST的性质来做。完全二叉树可以用一个数组 bst 存储,而BST的中序序列有序。题目已经说了是N个不同的整数,因此也不用考虑重复元素到底放左子树还是右子树了。对中序遍历的函数进行改写,在中序遍历到某节点时,按顺序将中序序列的元素赋给遍历到的节点。这样就完成了bst的赋值,输出数组 bst 就是 层序遍历的结果。
中序遍历时,要用到完全二叉树用数组存储所具有的性质:从下标0开始存储的话,若父亲节点下标为 i, 则左孩子下标为 2*i +1, 右孩子下标为 2 * i + 2, 同时注意中序遍历的条件:节点不为NULL, 对于数组而言就是下标小于数组长度。
本来还想按照层序遍历的规范写法来写的,后来才意识到直接输出 bst 数组 就是层序遍历的结果了......
AC代码如下:
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <cstring>
using namespace std;
void inOrder(vector<int> &in, vector<int> &tree, int parent)
{
static int i = 0;
if(i == in.size()) return;
if(parent < in.size())
{
inOrder(in, tree, parent * 2 + 1);
tree[parent] = in[i++];
inOrder(in, tree, parent * 2 + 2);
}
}
int main()
{
int N;
cin >> N;
vector<int> bst(N), v(N);
for (int i = 0; i < N; ++i)
{
scanf("%d", &v[i]);
}
sort(v.begin(), v.end());
inOrder(v, bst, 0);
for (int i = 0; i < N; ++i)
{
printf("%d", bst[i]);
if(i < N - 1) printf(" ");
}
return 0;
}