树查找
描述
有一棵树,输出某一深度的所有节点,有则输出这些节点,无则输出EMPTY。该树是完全二叉树。
输入描述:
输入有多组数据。 每组输入一个n(1<=n<=1000),然后将树中的这n个节点依次输入,再输入一个d代表深度。
输出描述:
输出该树中第d层得所有节点,节点间用空格隔开,最后一个节点后没有空格。
示例1
输入:
4
1 2 3 4
2
输出:
2 3
分析:
每i层的元素为2(i-1)~2i-1
#include <bits/stdc++.h>
using namespace std;
int n,x,flag;
int a[1010];
int main(){
while(cin>>n){
for(int i=1;i<=n;i++)
cin>>a[i];
cin>>x;
int sum=1;
while(x>1){
sum*=2;
x--;
}
if(sum<=n)
cout<<a[sum];
int t=sum*2;
if(sum>n) cout<<"EMPTY"<<endl;
else{
for(int i=sum+1;i<t&&i<=n;i++)
cout<<" "<<a[i];
}
cout<<endl;
}
return 0;
}