剑指 Offer -- 把二叉树打印成多行(六十)

最新推荐文章于 2020-08-07 21:30:31 发布

原创最新推荐文章于 2020-08-07 21:30:31 发布 · 156 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#剑指 Offer #C++ #算法 #数据结构

剑指 Offer 系列专栏收录该内容

65 篇文章

订阅专栏

本文介绍了一种使用BFS算法实现的二叉树层次遍历方法，通过队列保存每一层的节点，并将它们从左至右依次输出。每完成一层遍历，输出结果即为该层的所有节点值。

把二叉树打印成多行(六十)

题目描述:

从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

代码(已在牛客上 AC)

思路: BFS

class Solution {
public:
        vector<vector<int> > Print(TreeNode* root) {
            if (!root) return {};
            queue<TreeNode*> q;
            q.push(root);
            vector<vector<int>> res;
            while (!q.empty()) {
                int size = q.size();
                vector<int> cur(size);
                for (int i = 0; i < size; ++i) {
                    auto r = q.front();
                    q.pop();
                    cur[i] = r->val;
                    if (r->left) q.push(r->left);
                    if (r->right) q.push(r->right);
                }
                res.push_back(cur);
            }
            return res;
        }
};