九度OJ题目1176-树查找

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

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

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

样例输入：
4
1 2 3 4
2
样例输出：
2 3

参考代码：

/*完全二叉树的性质。*/

#include<cstdio>
#include<iostream>
#include<cmath>
using namespace std;
#define N 1001
int buf[N];
int main() {
	int n;

	while (scanf("%d", &n) != EOF) {
		for (int i = 1; i <=n; i++) scanf("%d", &buf[i]);
		int d;
		scanf("%d", &d);
		int s = pow(2, d - 1);
		int e = pow(2, d) - 1;
		int mark = 0;
		while (s <= n&&s <= e) {
			if (mark == 0) {
				printf("%d", buf[s]);
				mark = 1;
			}
			else
				printf(" %d", buf[s]);
			s++;
		}
		printf("\n");
	}
	return 0;
}