《算法通关村—原来如此简单》二叉树的层序遍历

来看一道经典的算法题 LeetCode102 : 二叉树的层序遍历

解题思路：

        我们可以使用一个队列 ，先将根节点root放进队列中，然后将根节点出队的同时，将它的左右节点入队。后面的节点操作也是如此，将该节点出队的之后，入队它的左右节点。具体实现我们可以使用一个while循环，结束条件为队列为空。

        当队列为空，就代表我们已经将所有元素都出队，入队了一遍，即遍历完了整棵二叉树。

执行过程：

        但是如何判断某一层访问完了呢？，我们可以使用一个变量size标记一下，size表示某一层的元素个数，只要出队，就将size减1，减到0就说明该层元素访问完了。当size变成0之后，这时队列中剩余元素的个数恰好就是下一层元素的个数，因此重新将size标记为下一层的元素个数就可以继续处理新的一行了，例如在上面的序列中： 

 首先3入队，队列的长度为1。我们将size赋值为队列的长度即size=1。

然后定义一个for循环，循环的次数为size的值。每次for循环中，我们执行的都是将该节点出队，然后入队其左右节点。

如：将3出队，队列长度变为0，然后入队它的左右节点，队列长度又变为2。

于是下一次for循环的循环次数为2，正好对应了下一层的节点数。

再来，我们将9，20出队，队列长度-2变为0，但是又入队了它们的左右节点，长度+4。也对应了下一层的节点数。

最后，我们把每层遍历到的节点都放入到一个结果集中，将其返回就可以了：

实现代码：

public List<List<Integer>> level102Order(TreeNode root) {
    if(root==null) {
      return new ArrayList<List<Integer>>();
    }

    List<List<Integer>> res = new ArrayList<List<Integer>>();
    LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
    //将根节点放入队列中，然后不断遍历队列
    queue.add(root);
    while(queue.size()>0) {
      //获取当前队列的长度，也就是当前这一层的元素个数
      int size = queue.size();
      ArrayList<Integer> tmp = new ArrayList<Integer>();
      //将队列中的元素都拿出来(也就是获取这一层的节点)，放到临时list中
      //如果节点的左/右子树不为空，也放入队列中
      for(int i=0;i<size;++i) {
        TreeNode t = queue.remove();
        tmp.add(t.val);
        if(t.left!=null) {
          queue.add(t.left);
        }
        if(t.right!=null) {
          queue.add(t.right);
        }
      }
      //此时的tmp就是当前层的全部元素，用List类型的tmp保存，加入最终结果集中
      res.add(tmp);
    }
    return res;
  }
}