leetcode102 Binary Tree Level Order Traversal(二叉树的层序遍历)

题目要求

给定一个二叉树，返回节点值的层序遍历结果。
（tips:从左到右，一层一层）

输入示例

For example:
给定一个二叉树 [3,9,20,null,null,15,7],
    3
   / \
  9  20
    /  \
   15   7
return 层序遍历结果如下:
[
  [3],
  [9,20],
  [15,7]
]

解题思路

根据输入示例的结果形式可知，我们需要两个vector来辅助解答。
一个vector存放每一层的遍历结果，一个vector存放最后的整体结果。
根据层序遍历的要求（从左到右，一层一层的特点），我们选用队列来实现（先进先出），试想一下，每次我们取队首元素，然后把它的值放入当前层的vector中，再把它的孩子节点放到队列里进行迭代，只要队列中还有元素我们就进行循环操作，直到到达叶子结点。
, 难点是：什么时候我们结束当前一层遍历去进行下一层呢？
答案：这里通过当前层中-队列-里元素的个数就能知道当前层进行几次循环取值操作,用for来进行循环取值。比如第二层有9，20两个元素，要进行两个for循环（注意这里有个大坑点，不要在for循环中使用q.size()因为这个队列q在for循环中是不断改变的）

主要代码（c++）

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        if(root==NULL) return {};
        vector<vector<int>>res; // 第一个vector
        queue<TreeNode*>q;
        q.push(root);
        while(!q.empty())// 迭代的条件
        {
            int size = q.size(); //这里要先保存下来队列的元素个数，不然在for中会改变
            vector<int>currentlevel; //第二个vector
            for(int i =0; i<size; ++i) //大坑点！ 你可以把size换成q.size()试试看
            {
                TreeNode * tmp = q.front(); // 下面是常规写法 记录每一层的值
                q.pop();
                currentlevel.push_back(tmp->val);
                if(tmp->left !=NULL) q.push(tmp->left);
                if(tmp->right !=NULL) q.push(tmp->right);
            } 
            res.push_back(currentlevel);
        }
        return res;
    }
};

主要代码python

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def levelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        queue = []
        res = []
        if root is None:
            return res
        queue.append(root)
        while len(queue)!=0:
            eachlevel = []
            size = len(queue)
            for i in range(size):
                temp = queue.pop(0)
                eachlevel.append(temp.val)
                if temp.left:
                    queue.append(temp.left)
                if temp.right:
                    queue.append(temp.right)
            res.append(eachlevel)
        return res

划重点：

有关层序遍历的迭代记得用 队列哈！！！

相似题目：
解答：leetcode107 Binary Tree Level Order Traversal 2 (二叉树的层序遍历2)

原题链接：https://leetcode.com/problems/binary-tree-level-order-traversal/