BFS

算法框架

参考来源:
作者:labladong
网址

问题的本质就是让你在一幅「图」中找到从起点 start 到终点 target 的最近距离,这个例子听起来很枯燥,
但是 BFS 算法问题其实都是在干这个事儿
* 本质上就是一幅「图」,让你从一个起点,走到终点,
* 问最短路径。这就是 BFS 的本质,框架搞清楚了直接默写就好。

记住下面这个框架就ok了:

# 计算起点start 到 终点 target 的最近距离
int BFS(Node start, Node target){
	Queue<Node> q; // 核心数据结构
	Set<Node> visited; // 避免走回头路

	q.offer(start); // 将起点加入队列
	visited.add(start);
	int step = 0; // 记录扩散的步数

	while(q not empty){
		int sz = q.size();
		/* 将当前队列中的所有节点向四周扩散 */
		for(int i = 0; i<sz; i++){
			Node cur = q.poll();
			/* 划重点:这里判断是否到达终点 */
			if(cur is target){
				return step;
			}
			/* 将 cur 的相邻节点加入队列 */
			for(Node x:cur.adj()){
				if(x not in visited){
					q.offer(x);
					visited.add(x);
				}
			}
		}
		/* 划重点:更新步数在这里 */
		step++;
	}
}
队列 q 就不说了,BFS 的核心数据结构;cur.adj() 泛指 cur 相邻的节点,比如说二维数组中,cur 上下左右四面的位置就是相邻节点;visited 的主要作用是防止走回头路,大部分时候都是必须的,但是像一般的二叉树结构,没有子节点到父节点的指针,不会走回头路就不需要 visited
二叉树的最小高度Leetcode111

在这里插入图片描述

怎么套到 BFS 的框架里呢?
首先明确一下起点 start 和终点 target 是什么,怎么判断到达了终点?
显然起点就是 root 根节点,终点就是最靠近根节点的那个「叶子节点」
if (cur.left == null && cur.right == null) 
    // 到达叶子节点
  • 那么,按照我们上述的框架稍加改造来写解法即可:
int minDepth(TreeNode* root){
	if(root == null) return 0;
	Queue<TreeNode*> q = new TreeNode();
	q->offer(root);
	# root 本身就是一层,depth 初始化为 1
	int depth = 1;
	
	while(!q->empty()){
		int sz = q->size();
		/* 将当前队列中的所有节点向四周扩散 */
		for(int i = 0; i<sz; i++){
			TreeNode* cur = q->poll();
			/* 判断是否到达终点 */
			if(cur->left == null &&cur->right==null){
				return depth;
			}
			/* 将 cur 的相邻节点加入队列 */
			if(cur->left!=null){
				q->offer(cur->left);
			}
			if(cur->right!=null){
				q->offer(cur->right);
			}
		}
		/* 这里增加步数 */
		depth++;
	}
	return depth;
}
  • 为什么 BFS 可以找到最短距离,DFS 不行吗?
首先,你看 BFS 的逻辑,depth 每增加一次,队列中的所有节点
都向前迈一步,这保证了第一次到达终点的时候,走的步数是最少的。
DFS 不能找最短路径吗?其实也是可以的,但是时间复杂度相对高很多。你想啊,DFS 实际上是靠递归的堆栈记录走过的路径,
你要找到最短路径,肯定得把二叉树中所有树杈都探索完才能对
比出最短的路径有多长对不对?而 BFS 借助队列做到一次一步「齐头并进」,是可以在不遍历完整棵树的条件下找到最短距离
的。
形象点说,DFS 是线,BFS 是面;DFS 是单打独斗,BFS 是集
体行动。这个应该比较容易理解吧
解开密码锁的最少次数LeetCode 题目是第 752 题
1. 第一步,我们不管所有的限制条件,不管 deadends 和 target 
的限制,就思考一个问题:如果让你设计一个算法,穷举所有可能的密码组合,你怎么做?
穷举呗,再简单一点,如果你只转一下锁,有几种可能?总共有 
4 个位置,每个位置可以向上转,也可以向下转,也就是有 8 种可能对吧

比如说从 "0000" 开始,转一次,可以穷举出 "1000", "9000", 
"0100", "0900"... 共 8 种密码。然后,再以这 8 种密码作为基础,对每个密码再转一下,穷举出所有可能...

仔细想想,这就可以抽象成一幅图,每个节点有 8 个相邻的节
点,又让你求最短距离,这不就是典型的 BFS 嘛,框架就可以
派上用场了,先写出一个「简陋」的 BFS 框架代码再说别的:

```cpp
// 将s[j]向上拨动一次
string plusOne(string s, int j){
	int n = s.length(); 
	char ch[n + 1];
	// copying the contents of the 
    // string to char array 
    strcpy(ch, s.c_str());
    if(ch[j]=='9')
    	ch[j]=='0';
   	else
   		ch[j]+=1;
	return new string(ch);
}

// 将s[j]向下拨动一次
string minusOne(string s, int j){
	int n = s.length(); 
	char ch[n + 1];
	// copying the contents of the 
    // string to char array 
    strcpy(ch, s.c_str());
    if(ch[j]=='0')
    	ch[j]=='9';
   	else
   		ch[j]-=1;
	return new string(ch);
}

// BFS 框架,打印出所有可能的密码
void BFS(string target){
	Queue<string> q = new string();
	q.offer("0000");
	
	while(!q->isEmpty()){
		int sz = q->size();
		/* 将当前队列中的所有节点向周围扩散 */
		for(int i =0; i<sz; i++){
			string cur = q.poll();
			/* 判断是否到达终点 */
			system.out.println(cur);
			
			/* 将一个节点的相邻节点加入队列 */
			for(int j = 0; j<4; j++){
				string up = plusOne(cur, j);
				string down = minusOne(cur, j);
				q.offer(up);
				q.offer(down);
			}
		}
		/* 在这里增加步数 */
	}
	return;
}
  • 这段 BFS 代码已经能够穷举所有可能的密码组合了,但是显然不能完成题目,有如下问题需要解决:
  1. 会走回头路。比如说我们从 “0000” 拨到 “1000”,但是等从队列拿出 “1000” 时,还会拨出一个 “0000”,这样的话会产生死循环
  2. 没有终止条件,按照题目要求,我们找到 target 就应该结束并返回拨动的次数
  3. 没有对 deadends 的处理,按道理这些「死亡密码」是不能出现的,也就是说你遇到这些密码的时候需要跳过
int openLock(string deadends, string target){
	set<string> deads = new HashSet<>();
	for(string s:deadends) deads.add(s);
	// 记录已经穷举过的密码,防止走回头路
	set<string> visited = new HashSet<>();
	Queue<string> q = new LinkedList<>();
	// 从起点开始启动广度优先搜索
	int step = 0;
	q.offer("0000");
	visited.add("0000");
	
	while(!q.isEmpty()){
		int sz=q.size();
		/* 将当前队列中的所有节点向周围扩散 */
		for(int i =o; i<sz; i++){
			string cur = q.poll();
			/* 判断是否到达终点 */
			if(deads.contains(cur)){
				continue; // 遇到deads跳过去不能加入队列
			}
			if(cur.equals(target)){
				return step;
			}
			
			/* 将一个节点的未遍历相邻节点加入队列 */
			for(int j = 0; j<4; j++){
				string up = plusOne(cur, j);
				if(!visited.contains(up)){
					// 没访问过才访问
					q.offer(up);
					visited.add(up);
				}
				string down = minusOne(cur, j);
				if(!visited.contains(down)){
					q.offer(down);
					visited.add(down);
				}
			}
		}
		/* 在这里增加步数 */
		step++;
	}
	// 如果穷举完都没找到目标密码,那就是找不到了
	return -1;
}
剑指13 机器人的运动范围
矩阵BFS总结几点:
1. 队列queue, 二维vector<<vector<bool>>>visited创建, 队首元素入队
2. 出队就是要访问了,访问是否满足题目条件:visited?格子没有限制?,既然已经入队肯定没有超过边界,访问过后记得打上标签
3. 出队元素邻居点的添加:在边界的范围内(邻居点的添加可以这样按照队首元素的访问一个个队首元素的添加,也可以目前队内一圈元素整个往外扩添加,后者可能更符合BFS的动画形象,但是前者不容易乱有条理)
4. 直到队列为空,或到达终点
// 方法一:
// 这个题跟BFS搜索有不同的地方在于,BFS的扩展是对队内所有元素扩展的
// 而本题有限制,即有些格子不能扩展,所以这就是不同
// 方法一,是先对队内元素单个遍历单个扩展,队内元素肯定是符合要求的
// 统一在出队且这个元素能访问之后,才进行step的更新和visited的更新
class Solution{
  public:
  int get_single_number(int x){
    int s = 0;
    while(x){
      s+=x%10;
      x/=10;
    }
    return s;
  }

  int get_sum(pair<int, int> p){
    return get_single_number(p.first)+get_single_number(p.second);
  }

  // BFS
  int movingCount(int m, int n, int k){
    queue<pair<int,int>> q;
    vector<vector<bool>> visited(m, vector<bool>(n));
    // 步数
    int step = 0;
    // start 入队
    q.push({0, 0});

    int dx[4]={-1,1,0,0}, dy[4]={0,0,-1,1};

    while(!q.empty()){
      // 1.栈顶元素出队
      auto t = q.front();
      q.pop();

      // 该出队元素是否能走呢?
      // 决定是否能走后再添加其邻居点
      if(get_sum(t) > k || visited[t.first][t.second]) continue; // 即使符合范围也不能走,不访问,下一个点入队

      // 能走
      step++;
      // 标记已访问
      visited[t.first][t.second]=true;

      // 2.处理完栈顶元素,再添加其邻居点
      for(int i = 0; i<4 ;i++){
        int x = t.first + dx[i], y = t.second+dy[i];
        
        // 是否在这个范围
        if(x>=0 && x<m && y>=0 && y<n){ // 
          q.push({x,y});
        }
      }
    }

    return step;
  }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值