北京邮电大学 树查找(java)

题目描述
有一棵树，输出某一深度的所有节点，有则输出这些节点，无则输出EMPTY。该树是完全二叉树。
输入描述:
输入有多组数据。
每组输入一个n(1<=n<=1000)，然后将树中的这n个节点依次输入，再输入一个d代表深度。
输出描述:
输出该树中第d层得所有节点，节点间用空格隔开，最后一个节点后没有空格。
示例1
输入
复制
4
1 2 3 4
2
输出
复制
2 3

import java.util.*;
import java.io.*;
import java.math.*;
import java.text.* ;
public class Main
{
	static char[] pre;
	static char[] in;
	public static void main(String[] args) {
		try {
			BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
			String str;
			while((str=br.readLine()) != null) {
				int n = Integer.parseInt(str);
				String[] parts = br.readLine().split(" ");	
				int m = Integer.parseInt(br.readLine());
				int len = (int)Math.pow(2, m-1);
				int start = len-1;
                if(start >= n) System.out.println("EMPTY");
                else {
                    System.out.print(parts[start]);
                    for(int i = start+1; i < n && i < len+start; i++) {
                        System.out.print(" "+parts[i]);
                    }
                    System.out.println();                    
                }
			}				 
		} catch(IOException e) {
			e.printStackTrace();
		}
	}
}