牛客网------KY194树查找

KY194 树查找

题目描述

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

输入描述:

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

输出描述:

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

示例1

输入

4
1 2 3 4
2

输出

2 3

题解：

#include <iostream>
#include <algorithm>
#include <cstring> 
#include <cmath>
using namespace std;
int main(){
    //利用数组存储完全二叉树
    int a[1024];
    memset(a, 0, sizeof(a));
    int t, d;
    cin>>t;
    for(int i=1; i<=t; i++){
        int x;
        cin>>x;
        a[i] = x;
    }
    cin>>d;
    //第d层起始节点
    int start = (int)pow(2, d-1);
    //end:第d层结尾节点+1
    int end = min((int)pow(2, d), t);
    if(end - start < 0){
        cout<<"EMPTY"<<endl;
    }else{
        int i = start;
        for(; i< end - 1; i++){
            cout<<a[i] <<" ";
        }       
        cout<<a[i]<<endl;
    }

    return 0;
}