树查找 九度教程第34题 完全二叉树的性质

题目链接

有一棵树，输出某一深度的所有节点，有则输出这些节点，无则输出EMPTY。该树是完全二叉树。
输入描述:
输入有多组数据。
每组输入一个n(1<=n<=1000)，然后将树中的这n个节点依次输入，再输入一个d代表深度。
输出描述:
输出该树中第d层得所有节点，节点间用空格隔开，最后一个节点后没有空格。
示例1
输入
4
1 2 3 4
2
输出
2 3

考察知识点：
完全二叉树的性质

AC代码：

#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
#define maxn 1002
int d(int x) {//返回结点所在的层数
	int tmp = 1;
	while (x > 1) {
		tmp++;
		x /= 2;
	}
	return tmp;
}
int num[maxn];
int main() {
	int n;
	while (cin >> n) {
		int t;
		for (int i = 1; i <= n; i++)cin >> num[i];
		int a = d(n);
		cin >> t;
		if (t > a) {
			cout << "EMPTY" << endl;
			continue;
		}
		int tmp = pow(2, t-1);
		int r = min(tmp * 2, n + 1);
		for (int i = tmp; i < r-1; i++) {
			cout << num[i] << " ";
		}
		cout << num[r - 1] << endl;
	}
}