[力扣每日习题][102]. 二叉树的层序遍历 2024.02.14

难度评级:中等

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

示例 1:

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

示例 2:

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

题解思路:DFS、先序遍历得到层序遍历序列

        本题初看,可以有基本思路,一般情形下,层序序列需要通过队列实现,根据队列先进先出的特性,将每层节点分别作为根节点遍历如下提供队列的代码模板。

代码如下:

    queue<int> que;
    levelorder(TreeNOde* root){
        if(root == NULL){
            return;
        }
        while(!que.empty()){
            TreeNode* front = que.front();
            que.pop();
            if(root->left){
                que.push(root->left);
            }
            if(root->right){
                que.push(root->right);
            }
        }
        res.push_back(root->val);
    }

        当然,我们根据返回数据类型是vector<vector<int>>,实际表现为{{},{},{}...},可以通过先序遍历得到层序遍历序列,我们可以定义一个deep记录当前节点的层数,用于调用vector里的vector下标,当deep大于等于res.size()时,需要扩充一个空vector,每层的节点对应每层的deep和vector,通过res[deep].push(root->val)索引,将对应节点压入对应的vector。下面通过图形更好理解

        先序遍历是根-左-右的序列顺序,所以会先遍历完左子树,再遍历右子树,就会得不到层序序列,所以递归的时候我们将deep一同递归,对应的节点就会压入对应的deep的vector中。

题解代码如下:

/**
 * Definition for a binary tree node.
 * 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>> res; // 定义返回数据类型
    void pre_DFS(TreeNode* root, int deep, vector<vector<int>>& res){
        if (root == NULL){
            return;
        }
        if (deep >= res.size()){
            res.push_back(vector<int> {});  // 扩容
        }
        res[deep].push_back(root->val);
        pre_DFS(root->left, deep + 1, res);  // 遍历左子树
        pre_DFS(root->right, deep + 1, res); // 遍历右子树
    }
    vector<vector<int>> levelOrder(TreeNode* root) {
        pre_DFS(root, 0, res);
        return res;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值