北邮OJ树查找

北邮OJ树查找

题目描述

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

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

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

示例1
输入
4
1 2 3 4
2
输出
2 3

#include<stdio.h>
#include<math.h> 
int main(){
	int n;
	while(scanf("%d",&n)!=EOF){
		int tree[1001];
		bool mark[1001];		//标记数组，判断此处是否为空节点 
		for(int i=1;i<=n;i++)
		mark[i]=false;
		for(int i=1;i<=n;i++){	//为方便计算，此处从1计数 
		scanf("%d",&tree[i]);	//输入节点 
		mark[i]=true;			//对所输入的节点进行标记 
	}
		int d;						//树层数
		scanf("%d",&d);
		int x=pow(2,d-1);		//求得d层第一个数的坐标 
		int y=pow(2,d)-1;		//求得d层节点总个数 
		for(int i=x;i<=y;i++){
			if(mark[x]==false){	//若d层第一个为空节点，则EMPTY 
				printf("EMPTY\n");
				break;
			}
			else{
				if(i==x) printf("%d",tree[i]);//处理空格 
				else{
					if(mark[i]==false) break;//判断是否为空节点，不空则输出 
					else
						printf(" %d",tree[i]);
				} 
			}
		}
}
	return 0;
} 

若担心pow()函数会产生精度问题可以用循环求得x,y，如下例。但此前已在牛客网提交通过。

int x=1;
while(d--){
	x=2*x;
}
int y=2*x-1;