LeetCode 102. 二叉树的层序遍历（DFS+BFS）

1.题目

给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。

示例：
二叉树：[3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回其层次遍历结果：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-tree-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.题解

DFS

深度优先搜索，需要记录元素的层数，之后按照层数大小进行插入结果数组。

class Solution
{
public:
    vector<vector<int>> levelOrder(TreeNode* root)
    {
        vector<vector<int>>res;
        dfs(res, root, 0);
        return res;
    }
    void dfs(vector<vector<int>> &res, TreeNode* root, int level)
    {
        if(!root) return;
        if(level >= res.size()) res.push_back(vector<int>());
        res[level].push_back(root -> val);
        dfs(res, root->left, level + 1);
        dfs(res, root->right, level + 1);
    }
};

BFS

官方题解，的确很棒，不需要维护深度
队列维护每一层的元素，从上到下，正好对应元素在数组中的位置，每出队一次就将其全放到二维数组中。

class Solution
{
public:
    vector<vector<int>> levelOrder(TreeNode* root)
    {
        vector<vector<int>>res;
        if(!root) return res;
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty())
        {
            int number = q.size();
            res.push_back(vector<int>());
            for(int i = 1; i <= number; i++)
            {
                auto node = q.front();
                q.pop();
                res.back().push_back(node -> val);
                if (node->left) q.push(node->left);
                if (node->right) q.push(node->right);
            }
        }
        return res;
    }
};