[leetcode]机器人的运动范围【BFS】

最新推荐文章于 2024-11-01 15:19:21 发布

Dtruth

最新推荐文章于 2024-11-01 15:19:21 发布

阅读量60

点赞数

文章标签： leetcode 宽度优先算法

本文链接：https://blog.csdn.net/m0_47464136/article/details/129594963

版权

该文章介绍了一个算法问题，通过使用一个mxn的map数组来表示地图，机器人能移动到的位置取决于坐标下标的数位和。算法采用广度优先搜索（BFS）从起点(0,0)开始，遍历所有符合条件（数位和小于K且未访问过）的格子，计算可以到达的格子数量。

摘要由CSDN通过智能技术生成

核心思想：使用一个mxn的map数组表示地图，由于机器人能否到达是看下标的数位和，因此可以就地使用map数组的值map[i][j]作为该点是否已经被访问的标志，首先将(0,0)点加入队列中，然后开始bfs遍历，点加入队列的条件为在地图中（不越界）以及下标数位和M<=K且未被访问过

class Solution {
public:
int dx[4]={1,-1,0,0};
int dy[4]={0,0,1,-1};
int getsum(int x)//求数位和
{
    int sum=0;
    while(x>0)
    {
        sum=sum+x%10;
        x=x/10;
    }
    return sum;
}
    int movingCount(int m, int n, int K) {
        int map[m][n];
	memset(map,0,sizeof(map));
	int reachcnt=1;
	queue<pair<int,int>> q;
	if(K==0)
	{
		return 1;
	}
	else{
		q.push({0,0});
		map[0][0]=1;
		while(!q.empty()){
			int topx=q.front().first;
			int topy=q.front().second;
			q.pop();
			for(int k=0;k<4;k++){
				int px=topx+dx[k];
				int py=topy+dy[k];
				if(px>=0&&px<m&&py>=0&&py<n&&getsum(px)+getsum(py)<=K&&map[px][py]==0){
					reachcnt++;
					q.push({px,py});
					map[px][py]=1;
				}
			}
		}
	}	
    return reachcnt;
    }
};