Hot100-Day48-T102二叉树的层序遍历

Day48[26/4/17]T102二叉树的层序遍历

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

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：[[3],[9,20],[15,7]]

示例 2：

输入：root = [1]
输出：[[1]]

示例 3：

输入：root = []
输出：[]

提示：

• 树中节点数目在范围 [0, 2000] 内
• -1000 <= Node.val <= 1000

解题思路

递归搜索的时候记录一下层数就可以了。

这个层数可以用一个变量来专门记录。

Code

#include <iostream>
#include <vector>

using namespace std;

struct TreeNode
{
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode() : val(0), left(nullptr), right(nullptr) {}
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
    TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
};

class Solution
{
public:
    vector<vector<int>> levelOrder(TreeNode *root)
    {
        // 0. 没有节点时
        if (root == nullptr)
        {
            return {};
        }

        vector<vector<int>> res;
        int layer = 0;
        DFS(root, res, layer);

        return res;
    }

private:
    void DFS(TreeNode *root, vector<vector<int>> &res, int layer)
    {
        if (root == nullptr)
        {
            return;
        }

        if (layer + 1 > res.size())
        {
            res.resize(layer + 1);
        }
        res[layer].push_back(root->val);

        DFS(root->left, res, layer + 1);
        DFS(root->right, res, layer + 1);
    }
};
auto main() -> int
{
    cout << "Hello World!\n"
         << endl;
}