从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7],
返回其层次遍历结果:
[
[3],
[9,20],
[15,7
]
提示:
节点总数 <= 1000
来源:力扣(LeetCode)
我的思路:这个题和层次遍历基本相同,但有些要修改的地方,解题思路如下:
java代码实现
import java.util.LinkedList;
import java.util.Queue;
import java.util.ArrayList;
import java.util.List;
public class Solution {
public static List<List<Integer>> levelOrder(BTree root)
{
Queue<BTree> queue = new LinkedList<BTree>();
List<List<Integer>> res = new ArrayList<List<Integer>>();
if(root != null) queue.add(root);
while(!queue.isEmpty())
{
List<Integer> ans = new ArrayList<Integer>();
for(int i = queue.size(); i>0; i--){//这个就是添加的循环
BTree a=queue.poll();
ans.add(a.data);
if(a.left!=null)
{
queue.add(a.left);
}
if(a.right!=null)
{
queue.add(a.right);
}
}
res.add(ans);
}
return res;
}
本题涉及到的知识:List和Arraylist
List是一个接口,是Collection接口的一个子接口,是一个有序的集合。
ArrayList是List的一个实现类,可以实现数组大小的可变,可以很方便的进行增加和删减数组内元素的操作。
为什么要用 List list = new ArrayList() ,而不用 ArrayList alist = new ArrayList()呢?
问题就在于List有多个实现类,现在你用的是ArrayList,也许哪一天你需要换成其它的实现类,如 LinkedList或者Vector等等,这时你只要改变这一行就行了:List list = new LinkedList(); 其它使用了list地方的代码根本不需要改动。
假设你开始用 ArrayList alist = new ArrayList(), 这下你有的改了,特别是如果你使用了ArrayList特有的方法和属性。
Arraylist 和Queue的如下图: