参考:http://www.voidcn.com/article/p-fltblucn-bmq.html
题目地址:https://leetcode.com/problems/binary-tree-level-order-traversal/
有三种解法:递归、队列、传统的BFS。
1.递归解法:
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<List<Integer>>();
levelRe(root,res,0);
return res;
}
public void levelRe(TreeNode root,List<List<Integer>> res,int level){
if(root==null)
return;
if(res.size()<level+1)
res.add(new ArrayList<Integer>());
res.get(level).add(root.val);
levelRe(root.left,res,level+1);
levelRe(root.right,res,level+1);
}
}
2.队列解法:
用队列的先进先出的机制将同一行的元素读入后,再逐个读出,并在同时插入下一行中的元素。
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> result = new ArrayList<List<Integer>>();
if(root==null)
return result;
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(root);
while(!queue.isEmpty()){
int levNum = queue.size();
List<Integer> temp = new ArrayList<Integer>();
for(int i = 0 ; i<levNum ; i++){
if(queue.peek().left!=null)queue.offer(queue.peek().left);
if(queue.peek().right!=null)queue.offer(queue.peek().right);
temp.add(queue.poll().val);
}
result.add(temp);
}
return result;
}
}
3.采用传统的BFS解法:
这里同样是维护一个队列,只是对于每个结点我们知道它的邻接点只有可能是左孩子和右孩子。
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<List<Integer>>();
if(root == null)
return res;
LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
queue.add(root);
int curNum = 0;
int lastNum = 1;
ArrayList<Integer> list = new ArrayList<Integer>();
while(!queue.isEmpty())
{
TreeNode cur = queue.poll();
lastNum--;
list.add(cur.val);
if(cur.left!=null)
{
queue.add(cur.left);
curNum ++;
}
if(cur.right!=null)
{
queue.add(cur.right);
curNum++;
}
if(lastNum==0)
{
lastNum = curNum;
curNum = 0;
res.add(list);
list = new ArrayList<Integer>();
}
}
return res;
}
}