算法模板之BFS

最新推荐文章于 2023-05-15 03:37:35 发布

Fighting_初心

最新推荐文章于 2023-05-15 03:37:35 发布

阅读量261

点赞数 1

分类专栏：算法竞赛

本文链接：https://blog.csdn.net/qq_34519487/article/details/104456797

版权

算法竞赛专栏收录该内容

5 篇文章 1 订阅

订阅专栏

一、基本模板

    void bfs(TreeNode* root, vector<vector<int>> &result, queue<TreeNode*> &q) {
        if(root == NULL) {  //判空
            return;
        }
        
        q.push(root);
   
        while(!q.empty()) {  //队列作为条件条件进行循环
            TreeNode* node = q.front();
            q.pop();

            if(node->left != NULL){
                q.push(node->left);
            } 
            if(node->right != NULL){
                q.push(node->right);
            }
        }
    }

二、常见写法举例

LeetCode 102. 二叉树的层次遍历

    void bfs(TreeNode* root, vector<vector<int>> &result, queue<TreeNode*> &q) {
        if(root == NULL) {
            return;
        }
        queue<int> dep;  //层  
        q.push(root);
        dep.push(0);
        while(!q.empty()) {
            TreeNode* node = q.front();
            q.pop();
            int d = dep.front(); 
            dep.pop();
            if(d == result.size()) {
                result.emplace_back();
            }
            result[d].push_back(node->val);
            if(node->left != NULL){
                q.push(node->left);
                dep.push(d + 1);
            } 
            if(node->right != NULL){
                q.push(node->right);
                dep.push(d + 1);
            }
        }
    }
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> result;
        queue<TreeNode*> q;
        bfs(root, result, q);
        return result;
    }