题目1176:树查找
时间限制:1 秒
内存限制:32 兆
特殊判题:否
-
题目描述:
-
有一棵树,输出某一深度的所有节点,有则输出这些节点,无则输出EMPTY。该树是完全二叉树。
-
输入:
-
输入有多组数据。
每组输入一个n(1<=n<=1000),然后将树中的这n个节点依次输入,再输入一个d代表深度。
-
输出:
-
输出该树中第d层得所有节点,节点间用空格隔开,最后一个节点后没有空格。
-
样例输入:
-
4 1 2 3 4 2
-
样例输出:
-
2 3
#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
****************************************************************/