算法学习②: BFS/广度优先遍历

BFS的思想通常可以帮助我们更好的理解树的结构,比如树的深度,是否是完全二叉树等.

注意,不同于DFS,处理BFS的问题时,通常我们更喜欢用的数据结构是queue==> 先进先出.

比如求二叉树的最大深度 leetcode链接

在这一题中,我们要能分辨不同层的节点,所用的方式就是在处理每一层的节点前,记录当前遍历层的node数(或者说是queue的长度).然后每次进行一个节点的处理,都会让计数器子减,当它变为0后,(此时queue不为空,因为下一层元素已经被存入)再去遍历,并且让层数加1.

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int maxDepth(TreeNode* root) {
        queue<TreeNode*> layer;
        if(root==nullptr) return 0;
        layer.push(root);
        int floors =0;
        int cnt_tmp=0;
        while(!layer.empty())
        {
            cnt_tmp= layer.size();
            while(cnt_tmp>0)
            {
                root = layer.front();
                layer.pop();
                if(root->left) layer.push(root->left);
                if(root->right) layer.push(root->right);
                cnt_tmp--;
            }
            floors++;
        }
        return floors;
    }
};

判断二叉树是否为完全二叉树 leetcode链接

这题用bfs的思想同样能很容易的解决,思路如下:

一棵树不是完全二叉树时,必然在树的左边存在零节点,换句话说,如果对于这棵树做BFS, 存入所有节点,不管是不是nullptr,当queue的最前面是nullptr时,我们终止搜索,如果这时候,queue的后面出现非零元素时,说明这棵数队头元素的同层或者下一层出现非零元素,那么这必然不是完全二叉树,

 /**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    bool isCompleteTree(TreeNode* root) {
        queue<TreeNode*> NodeQueue;
        NodeQueue.push(root);
        while(root)
        {
            NodeQueue.push(root->left);
            NodeQueue.push(root->right);
            NodeQueue.pop();
            root = NodeQueue.front();
        }
        while(!NodeQueue.empty())
        {
            root=NodeQueue.front();
            NodeQueue.pop();
            if(root)
            {
                return false;
            }
        }
        return true; 

    }
};

又比如Leetcode102 二叉树的层序遍历

为了保证每一层的序列放在一个vector排列，需要记录每一层的大小，对应下面代码中的size

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> res;
        if(!root) return res;
        //res.push_back(root);
        queue<TreeNode*> node_queue;
        node_queue.push(root);
        while(!node_queue.empty())
        {
            vector<int> tmp_vec;
            int size = node_queue.size();
            for(int i=0;i<size;i++)
            {
                TreeNode* head = node_queue.front();
                tmp_vec.push_back(head->val);
                node_queue.pop();
                if(head->left) node_queue.push(head->left);
                if(head->right) node_queue.push(head->right);
            }
            res.push_back(tmp_vec);
        }
        return res;

    }
};