:::info 💡宽度优先搜索(Breadth First Search,BFS ),也叫做广度优先搜索,或者层次遍历。一般在图论中叫宽度优先搜索或者广度优先搜索,在树结构中叫做层次遍历,目的都是遍历所有节点。 :::

图的广度优先搜索

BFS的做法是遍历当前节点的所有相邻节点,也就是下一层所有节点,然后在依次遍历下一层所有节点的相邻节点。如下图所示,当前节点为2号,那么相邻的3、5、4号节点为下一层节点,为了保证在遍历完4号节点之后,可以继续向下遍历,则需要暂存3、5、4号节点,并且可以取出继续分别访问它们的相邻节点。这里就需要使用数据结构中的队列,利用队列先进先出(Fisrt in Fisrt out, FIFO)的特点,可以对每一个节点访问其相邻节点,并且将相邻节点加入队列中,在遍历完当前层时,队列中充满的就是下一层节点。
宽度(广度)优先搜索 - 图1
在遍历的过程中还需要对已经访问的节点设置一个已经访问的标志,避免重复遍历进入死循环。
伪代码实现
用邻接矩阵的形式来表示图中节点的关系,在实际的问题中通常时二维矩阵,或者节点边的状态表示,可以将其转化为邻接矩阵。

  1. vector<int> visited; //记录节点是否访问
  2. queue<int> q; //队列
  3. q.insert(adj[0][0]); //先插入起始节点
  4. while (!q.empty()) { //当队列不为空时,继续搜索
  5. int n = q.size(); //当前层一共多少节点
  6. while (n--) { //搜索完当前层所有节点进入下一层
  7. int node = q.front();
  8. q.pop(); //出队
  9. auto &tmp = adj[node]; //当前节点的所有邻接节点
  10. for (auto &next:tmp) {
  11. if (visited[next]) continue; //若节点已经访问,则跳过
  12. visited[next]=true; //否则访问该节点,并且标记已经访问
  13. q.insert(next); //该节点加入队列
  14. }
  15. }

通过上述伪代码可以实现对图的宽度优先遍历,以离初状态的状态距离为序进行遍历。在实际问题中可以计算出图中节点之间的距离,比如力扣中的题目,计算网络的空闲时间,就可以通过计算各个节点离初状态的距离来计算网络中最远节点接受信息的时间。

树的层次遍历

对于树的层次遍历,思路跟图的广度优先遍历是相同的,在广度优先搜索的每一轮中,我们会遍历同一层的所有节点。
首先把根节点 root 放入队列中,随后在广度优先搜索的每一轮中,我们首先记录下当前队列中包含的节点个数(记为 cnt),即表示上一层的节点个数。在这之后,我们从队列中依次取出节点,直到取出了上一层的全部 cnt 个节点为止。当取出节点cur 时,我们将 cur 的值放入一个临时列表,再将cur 的所有子节点全部放入队列中。
当这一轮遍历完成后,临时列表中就存放了当前层所有节点的值。这样一来,当整个广度优先搜索完成后,我们就可以得到层序遍历的结果
宽度(广度)优先搜索 - 图2
以力扣的题目N叉树的层次遍历为例,如上图的二叉树为例,对N叉树进行层次遍历,并将结果存储在数组中,如[[1],[3,2,4],[5,6]]

  1. /*
  2. // Definition for a Node.
  3. class Node {
  4. public:
  5. int val;
  6. vector<Node*> children;
  7. Node() {}
  8. Node(int _val) {
  9. val = _val;
  10. }
  11. Node(int _val, vector<Node*> _children) {
  12. val = _val;
  13. children = _children;
  14. }
  15. };
  16. */
  17. class Solution {
  18. public:
  19. vector<vector<int>> levelOrder(Node* root) {
  20. if (root==nullptr) return {};
  21. vector<vector<int>> ans;
  22. queue<Node*> q;
  23. q.push(root);
  24. vector<int> tmp;
  25. while (!q.empty()) {
  26. int n = q.size();
  27. while (n--) {
  28. auto node = q.front();
  29. q.pop();
  30. tmp.emplace_back(node->val);
  31. auto childs = node->children;
  32. for (auto &child:childs) {
  33. q.push(child);
  34. }
  35. }
  36. ans.emplace_back(move(tmp));
  37. tmp.clear();
  38. }
  39. return ans;
  40. }
  41. };