102. Binary Tree Level Order Traversal

最新推荐文章于 2022-02-20 23:20:19 发布

zeroQiaoba

最新推荐文章于 2022-02-20 23:20:19 发布

阅读量214

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/zeroQiaoba/article/details/60771984

leetcode 专栏收录该内容

429 篇文章

订阅专栏

本文介绍了一种实现二叉树广度优先搜索(BFS)的方法，通过使用队列来跟踪节点及其深度，最终按层次顺序返回节点值。

简单树的BFS

/**
 * 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) {

        queue<TreeNode*> level;
        queue<int> depth;

        vector<vector<int>> result;

        if(root==NULL)
            return result;


        level.push(root);
        depth.push(1);

        while(!level.empty())
        {
            if(level.front()->left!=NULL)
            {
                level.push(level.front()->left);
                depth.push(depth.front()+1);

            }

            if(level.front()->right!=NULL)
            {
                level.push(level.front()->right);
                depth.push(depth.front()+1);
            }

            if(result.size()<depth.front())
            {
                vector<int> temp;
                temp.push_back(level.front()->val);
                result.push_back(temp);
            }
            else
                result[depth.front()-1].push_back(level.front()->val);

            level.pop();
            depth.pop();
        }

        return result;

    }
};