题目描述
地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格,但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如,当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。但是,它不能进入方格(35,38),因为3+5+3+8 = 19。请问该机器人能够达到多少个格子?
思路:先制作可进入的mapp数组地图,bfs遍历,并用cot记录数量返回。此题显然用bfs效率更高。
class Solution { public: int visit[1000][1000] = {0}; int mapp[1000][1000] = {0}; struct index{ int x; int y; }; int movingCount(int threshold, int rows, int cols) { if(threshold<0){ return 0; } for(int i=0;i<rows;i++){ for(int j=0;j<cols;j++){ int x = i; int y = j; int sum = 0; while(x||y){ sum += x%10; sum += y%10; x /= 10; y /= 10; } if(sum<=threshold){ mapp[i][j] = 1;//制作地图 } } } struct index v{0,0}; queue<index> q; q.push(v); int cot = 0; //开始bfs遍历,因为是从(0,0)开始右和下就可以了 while(!q.empty()){ int x = q.front().x; int y = q.front().y; if(x>=0&&x<rows&&y>=0&&y<cols){//先判断x,y是否越界 if(!visit[x][y]&&mapp[x][y]==1){ v.x = q.front().x + 1; v.y = q.front().y; q.push(v); v.x = q.front().x; v.y = q.front().y + 1; q.push(v); visit[x][y] = 1; cot++; } } q.pop(); } return cot; } };