KY194 树查找
题目描述
有一棵树,输出某一深度的所有节点,有则输出这些节点,无则输出EMPTY。该树是完全二叉树。
输入描述:
输入有多组数据。
每组输入一个n(1<=n<=1000),然后将树中的这n个节点依次输入,再输入一个d代表深度。
输出描述:
输出该树中第d层得所有节点,节点间用空格隔开,最后一个节点后没有空格。
示例1
输入
4
1 2 3 4
2
输出
2 3
题解:
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
int main(){
//利用数组存储完全二叉树
int a[1024];
memset(a, 0, sizeof(a));
int t, d;
cin>>t;
for(int i=1; i<=t; i++){
int x;
cin>>x;
a[i] = x;
}
cin>>d;
//第d层起始节点
int start = (int)pow(2, d-1);
//end:第d层结尾节点+1
int end = min((int)pow(2, d), t);
if(end - start < 0){
cout<<"EMPTY"<<endl;
}else{
int i = start;
for(; i< end - 1; i++){
cout<<a[i] <<" ";
}
cout<<a[i]<<endl;
}
return 0;
}