LintCode【简单】69. 二叉树的层次遍历。代码及思路

题目要求：

给出一棵二叉树，返回其节点值的层次遍历（逐层从左往右访问）

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给一棵二叉树 {3,9,20,#,#,15,7} ：

  3
 / \
9  20
  /  \
 15   7

返回他的分层遍历结果：

[
  [3],
  [9,20],
  [15,7]
]

思路：

层序遍历和前中后序遍历不一样，稍微复杂一些。思路注释在代码里了。

需要用到队列的知识，此处放一个链接：C++ 中queue（队列）的用法

代码：

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */

class Solution {
public:
    /*
     * @param root: A Tree
     * @return: Level order a list of lists of integer
     */
    vector<vector<int>> levelOrder(TreeNode * root) {
        // write your code here
        vector<vector<int>> result;//建立一个二维的vector最后用来返回
        if(root == NULL)  return result;//如果是空，则直接返回
        
        queue<TreeNode *> que;//建立一个存放树节点类型指针的队列que
        que.push(root);//放入root节点
        
        while(!que.empty()){//队列不是空（循环到队列为空结束）
            int size = que.size();//队列中元素的个数，表示这一层有几个元素
            vector<int> temp;
            //定义一层的vector容器，接下来循环将队列里前size个元素放入这一层，出队列
            for(int i = 0; i < size; i++){
                TreeNode *q = que.front();//这里不定义指针，直接用que.front()->val也可以，下同
                temp.push_back(q->val);
                que.pop();
                //判断有无左右孩子，放入队列
                if(q->left)  que.push(q->left);
                if(q->right) que.push(q->right);
            }
            //循环结束后将这一层放入result
            result.push_back(temp);
        }
        return result;
    }
};