LeetCode. 102 二叉树的层次遍历

最新推荐文章于 2021-02-25 17:39:45 发布

原创最新推荐文章于 2021-02-25 17:39:45 发布 · 636 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #层序遍历

C++ 同时被 3 个专栏收录

262 篇文章

订阅专栏

LeetCode

94 篇文章

订阅专栏

LeetCode

94 篇文章

订阅专栏

本文介绍了一种使用BFS遍历方法实现的二叉树层次遍历算法。通过将节点按层压入队列，并利用队列中元素数量来记录每一层节点的数量，最终实现了从上至下、从左至右的层次遍历。

给定一个二叉树，返回其按层次遍历的节点值。（即逐层地，从左到右访问所有节点）。

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

返回其层次遍历结果：

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

本题我是利用BFS遍历来实现，按层将节点不断压入队列，因为同一层的节点是在一个vector中的。所以还需要记录节点所在层数。这里恰好可以利用“当前层的节点个数为队列中元素个数”这个性质。

代码展示：

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> ans;
        queue<TreeNode*> q;
        if(root==NULL)
            return ans;
        q.push(root);
        while(!q.empty()){
            vector<int> temp;
            int size = q.size();
            while(size>0){
                TreeNode* node = q.front();
                temp.push_back(node->val);
                q.pop();
                if(node->left!=NULL)
                    q.push(node->left);
                if(node->right!=NULL)
                    q.push(node->right);
                size -= 1;
            }
            ans.push_back(temp);
        }
        return ans;
    }
};