102. 二叉树的层序遍历

题目描述:

给你二叉树的根节点 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

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

分析:

层序遍历很经典了,基本上离不开队列。通常的步骤是:

        1.根节点入队列

        2.在队列不空的情况下开始循环

        3.取出队列头元素

        4.将其左右节点入队列

但是这道题有一些小改变,要求将每一层的节点存放在一起,以此来体现树的层次。因此,需要做一些调整:

        1.根节点入队列q

        2.在队列q不空的情况下开始循环

        3.取出队列q中所有元素

        4.队列q中所有元素的左右节点存入新队列tempq

        5.交换新队列tempq和队列q,返回步骤2

代码如下:

/**
 * 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;
    // 存储节点的队列
    queue<TreeNode *> q;
    vector<vector<int>> levelOrder(TreeNode* root) {
        // 空树直接返回res
        if(!root) return res;
        // 根节点入队列
        q.push(root);
        // 队列不空的情况下,进入循环
        while(!q.empty())
        {
            // tempq,用来存储q中节点的左右子节点
            queue<TreeNode *> tempq;
            // tempres用来存放本层的层序遍历序列
            vector<int> tempres;
            // q不空的情况下,清空q里的所有元素
            while(!q.empty())
            {
                // q中节点出队列
                TreeNode *temp=q.front();
                q.pop();
                // 存入tempres
                if(temp) tempres.push_back(temp->val);
                else continue;
                // 左右节点存入tempq
                tempq.push(temp->left);
                tempq.push(temp->right);
            }
            // 如果tempres不为空,则将tempres存入res
            if(tempres.size()>0)
            res.push_back(tempres);
            // 交换q和tempq 此时q为空,tempq中存放了下一层要访问的节点
            swap(q,tempq);
           
        }
        return res;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值