-
Notifications
You must be signed in to change notification settings - Fork 32
/
Copy path52_levelOrder.java
70 lines (65 loc) · 2.2 KB
/
52_levelOrder.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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
public class Solution {
/**
* @param root: The root of binary tree.
* @return: Level order a list of lists of integer
*/
//迭代实现
public ArrayList<ArrayList<Integer>> levelOrder(TreeNode root) {
// write your code here
ArrayList<ArrayList<Integer>> traverResult = new ArrayList<ArrayList<Integer>>();
if(root==null)
return traverResult;
ArrayList<TreeNode> nodeQueue = new ArrayList<TreeNode>();
nodeQueue.add(root);
while(nodeQueue.size()!=0){
traverResult.add(getSeriesOfLevel(nodeQueue));
ArrayList<TreeNode>tmpList = new ArrayList<TreeNode>();
for(TreeNode node:nodeQueue){
if(node.left!=null)
tmpList.add(node.left);
if(node.right!=null)
tmpList.add(node.right);
}
nodeQueue.removeAll(nodeQueue);
nodeQueue.addAll(tmpList);
}
return traverResult;
}
public ArrayList<Integer> getSeriesOfLevel(ArrayList<TreeNode> level){
ArrayList<Integer> series = new ArrayList<Integer>();
for(TreeNode node:level)
series.add(node.val);
return series;
}
//递归实现
public class Solution {
/**
* @param root: The root of binary tree.
* @return: Level order a list of lists of integer
*/
public ArrayList<ArrayList<Integer>> levelOrder(TreeNode root) {
// write your code here
ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
if(root==null)
return list;
ArrayList<TreeNode> tmp = new ArrayList<TreeNode>();
tmp.add(root);
traverTheTree(tmp,list);
}
public void traverTheTree(List<TreeNode> list,ArrayList<ArrayList<Integer>> result){
ArrayList<Integer> tmp = new ArrayList<Integer>();
ArrayList<TreeNode> tmpNode = new ArrayList<TreeNode>();
for(int i=0; i<list.size(); i++){
TreeNode node = list.get(i);
if(node.left!=null)
tmpNode.add(node.left);
if(node.right!=null)
tmpNode.add(node.right);
tmp.add(node.val);
}
if(tmp.size()>0)
result.add(tmp);
traverTheTree(tmpNode,result);
}
}
}