【机试备考】Day8-树查找

题目

牛客网-2010BUPT网研第四题
有一棵树，输出某一深度的所有节点，有则输出这些节点，无则输出EMPTY。该树是完全二叉树。

输入描述

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

输出描述

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

示例

题解

因为是完全二叉树，所以只需知道对应层的序号输出即可，大概只是想考考树的性质吧

#include<iostream>
#include <math.h>
using namespace std;
int main()
{
    
    
    int n;
    while(cin>>n)
    {
    
    
        //输入
        int d,tree[1001];
        for(int i=0;i<n;i++)
            cin>>tree[i];
        cin>>d;
        //没有,则输出EMPTY
        if(n<pow(2,d-1)-1)
        {
    
    
            cout<<"EMPTY";
            break;
        }
        //有,则输出对应层的节点序号
        int start=pow(2,d-1)-1;//第d层的起始节点序号
        int i;
        for(i=start;i<n-1&&i<start+pow(2,d-1)-1;i++)
            cout<<tree[i]<<" ";
        cout<<tree[i];
    }
}