【北邮机试刷题】树查找

时间限制：1秒  空间限制：65536K  热度指数：3162

 算法知识视频讲解

题目描述

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

输入描述:

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

输出描述:

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

示例1

输入

4
1 2 3 4
2

输出

2 3

package test;

import java.util.Scanner;

public class SearchTree {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner sc=new Scanner(System.in);
		while (sc.hasNext()) {
			int n=sc.nextInt();
			int[] tree=new int[n];
			for (int i = 0; i < n; i++) {
				tree[i]=sc.nextInt();
			}
			int d=sc.nextInt();
			int begin=(int)Math.pow(2, d-1)-1;
			int end=(int)Math.pow(2, d)-2;
			if (begin>n) {//判断有没有当前层
				System.out.println("EMPTY");
			}
			else {
				if (end>n) {
					while (begin<n-1) {
						System.out.print(tree[begin]+" ");
						begin++;
					}
					System.out.println(tree[n-1]);
				}else {
					while (begin<end) {
						System.out.print(tree[begin]+" ");
						begin++;
					}
					System.out.println(tree[end]);
				}
			}
		}
	}

}