九度OJ题目1176：树查找

题目1176：树查找

时间限制：1 秒

内存限制：32 兆

特殊判题：否

题目描述：

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

输入：

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

输出：

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

样例输入：

4
1 2 3 4
2

样例输出：

2 3

来源：

2010年北京邮电大学网院研究生机试真题

#include <stdio.h>
#include <math.h>
int main(){
	int n,d;
	int node[1010];
	while(scanf("%d",&n)!=EOF){ //输入n
		int h=(int)(log(n)/log(2)+1);//h为该完全二叉树的深度
		for(int i=1;i<=n;i++){
			scanf("%d",&node[i]);//输入树中的n个节点
		}
		scanf("%d",&d);//输入d
		if(d==1)	
			printf("%d\n",node[1]);//如果是第一层，就输出根节点
		else if(d>=2&&d<h){ //如果是中间层，则均为满层
			int high=pow(2,d)-1,low=pow(2,d-1);//计算节点下标
			for(int j=low;j<high;j++)
				printf("%d ",node[j]);//节点间用空格隔开
				printf("%d\n",node[high]);//最后一个节点后没有空格，输出换行
		}
		else if(d==h){ //如果是最后一层
			int s=pow(2,h-1);//计算本层节点初始下标
			for(int k=s;k<n;k++)
				printf("%d ",node[k]);//节点间用空格隔开
			printf("%d\n",node[n]);//最后一个节点后没有空格，输出换行
		}
		else
			printf("EMPTY\n");//没有就输出EMPTY并换行
	}
	return 0;
}
/************************************************************** 
    Problem: 1176 
    User: zpy
    Language: C++ 
    Result: Accepted 
    Time:0 ms 
    Memory:1128 kb 
****************************************************************/