完全二叉树查找

描述

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

输入描述

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

输出描述

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

用例输入 1

5
1 2 3 4 5 
7
7
1 2 3 4 5 6 7 
2
0

用例输出 1

EMPTY
2 3

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int a;
    while (cin>>a)
    {
        if (a==0) break;
        int li[a][a];
        for (int i=0;i<a;i++)
        {
            for (int j=0;j<a;j++)
            {
                li[i][j]=0;
            }
        }        
        int n;
        int h=0,l=0,s=1,k=0;
        for (int i=0;i<a;i++)
        {
            cin>>n;
            if (k==s) 
            {
                h++;
                l=0;
                s*=2;
                k=0;
            }
            li[h][l]=n;
            l++;
            k++;
        }
        int b;
        cin>>b;
        if (b>h) cout<<"EMPTY"<<endl;
        else
        {
            int q=1;
            for (int i=0;i<b-1;i++)
            {
                q*=2;
            }
            for (int i=0;i<q;i++)
            {
                cout<<li[b-1][i]<<" ";
            }
            cout<<endl;
        }
    }
}

//关注一下吧