题目描述:
有一棵树,输出某一深度的所有节点,有则输出这些节点,无则输出EMPTY。该树是完全二叉树。
输入:
输入有多组数据。
每组输入一个n(1<=n<=1000),然后将树中的这n个节点依次输入,再输入一个d代表深度。
输出:
输出该树中第d层得所有节点,节点间用空格隔开,最后一个节点后没有空格。
样例输入:
4
1 2 3 4
2
样例输出:
2 3
#include <iostream>
#include <cstdio>
#include <cmath>
#define MAXSIZE 500
using namespace std;
typedef struct BiNode{
int lc;
int rc;
int data;
bool turnOn;
BiNode(){//initiate automatically except data
initNode();
}
void initNode(){
turnOn=false;
lc=rc=-1;
}
}*BiTree;
BiNode Tree[MAXSIZE];
int cursor;
//非指定下标分配结点
int createNode(int data){//return the index of new node
if (cursor+1==MAXSIZE)
return -1;//failed
int nowP=cursor;
Tree[nowP].turnOn=true;
cursor++;
Tree[nowP].data=data;
Tree[nowP].lc=Tree[nowP].rc=-1;
return nowP;
}
//按下标分配结点
int createNode(int index,int data){//return the index of new node
Tree[index].turnOn=true;
Tree[index].data=data;
Tree[index].lc=Tree[index].rc=-1;
return index;
}
void freeNode(int index){
Tree[index].initNode();
}
void freeTree(int index){//postOrder
if (index!=-1){
//traverse
freeTree(Tree[index].lc);
freeTree(Tree[index].rc);
//visit
Tree[index].initNode();
}
}
int main(){
int n,d;
int data;
while (scanf("%d",&n)!=EOF){
//initiate
for (int i=0;i<MAXSIZE;i++){
Tree[i].initNode();
}
cursor=0;
//input
for (int i=0;i<n;i++){
scanf("%d",&data);
createNode(i,data);
}
scanf("%d",&d);
//output
int start=pow(2,d-1)-1;
int end=pow(2,d)-1-1;
bool firstCase=true;
int i;
for (i=start;i<=end;i++){
if (Tree[i].turnOn==false)
break;
//else output
if (firstCase)
firstCase=false;
else
printf(" ");
printf("%d",Tree[i].data);
}
if (i==start)
printf("EMPTY");
printf("\n");
}
return true;
}