A1064题目将完全二叉树与二叉搜索树结合起来,给定一个序列,要求输出其完全二叉搜索树的层序遍历。
思路很重要:
- 完全二叉树使用一个数组来存储
- 对于一棵BST来说,其中序遍历序列是递增的
- 所以,先将给出的原始序列进行递增排序
- 然后对表示完全二叉树的数组进行中序遍历,在遍历过程中填入排序后的原始序列
- 最后数组中存放的就是一棵完全二叉搜索树,存放顺序就是层序遍历的顺序。
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1001;
int CBT[maxn] = {0}, ori[maxn];
int n, p = 0;
void inOrder(int a[], int index){
if(index > n) return;
inOrder(a, index * 2);
a[index] = ori[p++];
inOrder(a, index * 2 + 1);
}
int main(){
cin >> n;
for(int i = 0; i < n; i++){
scanf("%d", &ori[i]);
}
sort(ori, ori + n);
inOrder(CBT, 1);
for(int i = 1; i <= n; i++){
printf("%d", CBT[i]);
if(i < n){
printf(" ");
}else{
printf("\n");
}
}
return 0;
}
A1099题目思路是一样的,只不过树的结构不是完全二叉树,而是给定的一种形态。同样是利用BST的中序遍历是有序序列这个性质,在中序遍历树的时候把从小到大排序好的序列填进去即可。
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
const int maxn = 101;
struct node{
int data;
int lchild, rchild;
}Node[maxn];
int N, seq[maxn];
void initial(){
for(int i = 0; i < maxn; i++){
Node[i].lchild = Node[i].rchild = -1;
}
}
int index = 0;
void inOrder(int root){
if(root == -1) return;
inOrder(Node[root].lchild);
Node[root].data = seq[index++];
inOrder(Node[root].rchild);
}
void layerOrder(int root){
queue<int> q;
q.push(root);
int cnt = 0;
while(!q.empty()){
int front = q.front();
q.pop();
printf("%d", Node[front].data);
cnt++;
if(cnt < N) printf(" ");
else printf("\n");
if(Node[front].lchild != -1){
q.push(Node[front].lchild);
}
if(Node[front].rchild != -1){
q.push(Node[front].rchild);
}
}
}
int main(){
initial();
cin >> N;
for(int i = 0; i < N; i++){
scanf("%d %d", &Node[i].lchild, &Node[i].rchild);
}
for(int i = 0; i < N; i++){
scanf("%d", &seq[i]);
}
sort(seq, seq + N);
inOrder(0);
layerOrder(0);
return 0;
}