问题 A: 树查找
时间限制: 1 Sec 内存限制: 32 MB
提交: 221 解决: 127
[提交][状态][讨论版][命题人:外部导入]
题目描述
有一棵树,输出某一深度的所有节点,有则输出这些节点,无则输出EMPTY。该树是完全二叉树。
输入
输入有多组数据。
每组输入一个n(1<=n<=1000),然后将树中的这n个节点依次输入,再输入一个d代表深度。
输出
输出该树中第d层得所有节点,节点间用空格隔开,最后一个节点后没有空格。
样例输入
5 1 2 3 4 5 7 7 1 2 3 4 5 6 7 2 0
样例输出
EMPTY 2 3
#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=1010;
int main(){
int n;
while(scanf("%d",&n)&&n){
int a[n];
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
}
int h;
scanf("%d",&h);
if(n<pow(2,(h-1))){
printf("EMPTY\n");
}
else {
int lo=pow(2,(h-1));
int hi=n<pow(2,h)-1 ? n:pow(2,h)-1 ;
for(int j=lo;j<=hi;j++){
if(j<hi)printf("%d ",a[j-1]);
else printf("%d\n",a[j-1]);
}
}
}
return 0;
}