原题链接:
http://oj.leetcode.com/problems/binary-tree-level-order-traversal/
这道题要求实现树的层序遍历,其实本质就是把树看成一个有向图,然后进行一次 广度优先搜索 ,这个图遍历算法是非常常见的,这里同样是维护一个队列,只是对于每个结点我们知道它的邻接点只有可能是左孩子和右孩子,具体就不仔细介绍了。算法的复杂度是就结点的数量,O(n),空间复杂度是一层的结点数,也是O(n)。代码如下:
这道题要求实现树的层序遍历,其实本质就是把树看成一个有向图,然后进行一次 广度优先搜索 ,这个图遍历算法是非常常见的,这里同样是维护一个队列,只是对于每个结点我们知道它的邻接点只有可能是左孩子和右孩子,具体就不仔细介绍了。算法的复杂度是就结点的数量,O(n),空间复杂度是一层的结点数,也是O(n)。代码如下:
public ArrayList<ArrayList<Integer>> levelOrder(TreeNode root) {
ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<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;
}
层序遍历也是树的一种遍历方式,在某些题目中会比较实用,不过这样其实更接近于图的问题了,在有些树的题目中层序遍历提供了更好的方法,所以还是得熟悉哈。层序遍历也是树的一种遍历方式,在某些题目中会比较实用,不过这样其实更接近于图的问题了,在有些树的题目中层序遍历提供了更好的方法,所以还是得熟悉哈。这道题还有一个变体Binary Tree Zigzag Level Order Traversal,其实也是进行广度优先搜索,不过因为要求不同,要换一种数据结构来记录层节点,有兴趣可以看看。