树查找

题目描述

有一棵树，输出某一深度的所有节点，有则输出这些节点，无则输出EMPTY。该树是完全二叉树。

输入描述:

输入有多组数据。
每组输入一个n(1<=n<=1000)，然后将树中的这n个节点依次输入，再输入一个d代表深度。

输出描述:

输出该树中第d层得所有节点，节点间用空格隔开，最后一个节点后没有空格。

示例1

输入

4
1 2 3 4
2

输出

2 3

//根据完全二叉树的性质，每一层的深度为depth，那么该层的结点的个数为(2^depth)-1;

#include<iostream>

#include<stdio.h>
#include<math.h>
using namespace std;
int main()
{
    int n,a[1001];
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=0;i<n;i++)
        {
            cin>>a[i];
        }
        int depth;
        cin>>depth;
        int count=pow(2,depth-1)-1;
        if(count>n)cout<<"EMPTY"<<endl; //这句话的意思就是当上一层的count都比n大的话，那么这一层的结点一定为空。
        else{
            for(int i=count;i<count+pow(2,depth-1);i++)
        {
            if(i==count)cout<<a[i];
            else cout<<' '<<a[i];
        }
        cout<<endl;
     }


    }
}