树查找

题目描述

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

输入描述:

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

输出描述:

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

示例1

输入

复制

4
1 2 3 4
2

输出

复制

2 3

#include <iostream>
#include <cstdio>
#include <cmath>
using namespace std;

int main()
{
    int n;
    int a[10001];

    while(~scanf("%d", &n))
    {
        for(int i = 1; i <= n; i++)
            scanf("%d", &a[i]);
        int k;
        scanf("%d", &k);
        int l1 = (int)pow(2,k-1);
        int l2 = min(n,(int)pow(2,k));
        if((int)pow(2,k-1)+1 > n)
        {
            printf("EMPTY");continue;
        }
        for(int i = l1; i < l2-1; i++)
            printf("%d ",a[i]);
        printf("%d\n",a[l2-1]);
    }

    return 0;
}