问题 A: 树查找
题目描述
有一棵树,输出某一深度的所有节点,有则输出这些节点,无则输出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 <cstdio>
#include <cmath>
using namespace std;
int t[1010];
int main()
{
int n,d;
while(scanf("%d",&n)&&n!=0)
{
for(int i=1;i<=n;++i)
{
scanf("%d",&t[i]);
}
scanf("%d",&d);
if(n<pow(2,d-1))
{
printf("EMPTY\n");
}
else
{
//完全二叉树的上层最后一个结点编号为pow(2,d-1)-1(高度为H的二叉树至多有2^H-1个结点)
//则下层第一个结点为pow(2,d-1)
int low=pow(2,d-1);
int high=n<pow(2,d)-1?n:pow(2,d)-1;
for(int i=low;i<=high;++i)
{
printf("%d",t[i]);
if(i<high)
printf(" ");
}
printf("\n");
}
}
}