剑指 Offer 32 - II. 从上到下打印二叉树 II

最新推荐文章于 2023-08-01 20:07:52 发布

允谦呀

最新推荐文章于 2023-08-01 20:07:52 发布

阅读量247

点赞数

分类专栏：剑指Offer 文章标签： leetcode 深度优先动态规划 c++ 算法

本文链接：https://blog.csdn.net/qiangqiang103/article/details/123154355

版权

剑指Offer 专栏收录该内容

24 篇文章 0 订阅

订阅专栏

剑指 Offer 32 - II. 从上到下打印二叉树 II

题目链接：题目链接

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

例如:
给定二叉树: [3,9,20,null,null,15,7],

返回其层次遍历结果：

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

提示：

节点总数 <= 1000

思路：

本题考察的是BFS(广度优先遍历)算法：我们借助队列的先进先出的特性，用队列存储上二叉树上的每个结点，借助一个vector数组存储结点上的值。

首先特判：当根结点为空时，直接返回[]

层次遍历+控制内层循环

内层：当一层中所有数都遍历一遍结束循环,遍历的同时将下一层的数加入到队列中

外层：当所有层遍历一遍结束循环

/**
 * 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) {
        vector<vector<int>> res;
        queue<TreeNode *> q;
        q.push(root);
        if(!root) return res;
        while(!q.empty()){
            int col = q.size();
            vector<int> tmp;
            while(col--){
                TreeNode *p = q.front();
                q.pop();
                tmp.push_back(p->val);
                if(p->left!=NULL) q.push(p->left);
                if(p->right!= NULL) q.push(p->right);
            }
            res.push_back(tmp);
        }
        return res;
    }
};