树查找
题目描述:
有一棵树,输出某一深度的所有节点,有则输出这些节点,无则输出EMPTY。该树是完全二叉树。
输入:
输入有多组数据。
每组输入一个n(1<=n<=1000),然后将树中的这n个节点依次输入,再输入一个d代表深度。
输出:
输出该树中第d层得所有节点,节点间用空格隔开,最后一个节点后没有空格。
样例输入:
4
1 2 3 4
2
样例输出:
2 3
Code
#include<iostream>
#include<cmath>
using namespace std;
int main()
{
int n, a[1001];
while (cin >> n)
{
for (int i = 1; i <= n; i++)
cin >> a[i];
//计算层数
int depth = floor((log(n) / log(2))) + 1;
int d;
cin >> d;
//分情况讨论
if (d == depth)
{
for (int i = (int)pow(2, d - 1); i <= n; i++)
{
if (i == (int)pow(2, d - 1))
cout << a[i];
else
cout << " " << a[i];
}
}
else if (d < depth)
{
for (int i = (int)pow(2, d - 1); i < (int)pow(2, d); i++)
{
if (i == (int)pow(2, d - 1))
cout << a[i];
else
cout << " " << a[i];
}
}
else
cout << "EMPTY" << endl;
cout << endl;
}
return 0;
}