完全二叉树查找

描述

有一棵树，输出某一深度的所有节点，有则输出这些节点，无则输出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;}}
}