九度OJ - 1176 - 树查找

题目描述

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

输入

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

输出

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

样例输入

4
1 2 3 4
2

样例输出

2 3

来源

2010年北京邮电大学网院研究生机试真题

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

int tree[1005]; 

int main()
{
    int n, m;
    while(scanf("%d", &n)!=EOF){
        for(int i = 1; i <= n; i++)
            scanf("%d", &tree[i]);

        cin >> m;

        int beg = (int)pow(2, m-1);
        int end = (int)pow(2, m);

        if(beg > n){
            cout << "EMPTY" << endl;
        }
        else{
            for(int i = beg; i < end && i <= n; i++){
                if(i-beg)
                    cout << " ";
                cout << tree[i];
            }
            cout << endl;
        }

    }
    return 0;
}