编程练习
题目:从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
思路:很简单,比前一道之字形打印还少了一步偶数层反转的步骤。
java代码如下:1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
public class PrintByLevel {
static ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> reslist=new ArrayList<ArrayList<Integer>>();
Queue<TreeNode> treeQueue=new LinkedList<TreeNode>();
if (pRoot==null) {
return reslist;
}
int level=0;
treeQueue.add(pRoot);
while(!treeQueue.isEmpty()) {
level++;
int len=treeQueue.size();
ArrayList<Integer> list=new ArrayList<Integer>();
for(int i=0;i<len;i++) {
TreeNode node=treeQueue.poll();
list.add(node.val);
if(node.left!=null) {
treeQueue.add(node.left);
}
if(node.right!=null) {
treeQueue.add(node.right);
}
}
reslist.add(list);
}
return reslist;
}
public static void main(String[] args) {
TreeNode root=new TreeNode(0);
TreeNode node1=new TreeNode(1);
TreeNode node2=new TreeNode(2);
TreeNode node3=new TreeNode(3);
TreeNode node4=new TreeNode(4);
TreeNode node5=new TreeNode(5);
root.left=node1;
root.right=node2;
node1.left=node3;
node1.right=node4;
node2.left=node5;
System.out.println(Print(root));
}
}