树查找

树查找

题目描述：

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

输入：

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

输出：

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

样例输入：

4
1 2 3 4
2

样例输出：

2 3

code

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

int Node[1001];

int main()
{
    int n, d;
    while (cin >> n && n > 0)
    {
        for (int i = 1; i <= n; i++)
            cin >> Node[i];
        cin >> d;

        if (n < pow(2, d - 1) - 1)
            cout << "EMPTY" << endl;

        else if (n>pow(2, d) - 1)
        {
            for (int j = (int)pow(2, d - 1); j <= pow(2, d) - 1; j++)
            {
                if (j == (int)pow(2, d - 1))
                    cout << Node[j];
                else
                    cout << " " << Node[j];
            }
        }

        else
        {
            for (int j = (int)pow(2, d - 1); j <= n; j++)
            {
                if (j == (int)pow(2, d - 1))
                    cout << Node[j];
                else
                    cout << " " << Node[j];
            }
        }
        cout << endl;
    }

    return 0;
}