地上有一个m行n列的方格,从坐标 [0,0] 到坐标 [m-1,n-1] 。一个机器人从坐标 [0, 0] 的格子开始移动,它每次可以向左、右、上、下移动一格(不能移动到方格外),也不能进入行坐标和列坐标的数位之和大于k的格子。例如,当k为18时,机器人能够进入方格 [35, 37] ,因为3+5+3+7=18。但它不能进入方格 [35, 38],因为3+5+3+8=19。请问该机器人能够到达多少个格子?
示例 1:
输入:m = 2, n = 3, k = 1
输出:3
示例 2:
输入:m = 3, n = 1, k = 0
输出:1
提示:
1 <= n,m <= 100
0 <= k <= 20
链接:https://leetcode-cn.com/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof
跟前一题类似,也需要建一个数组去存储访问过的位置。需要注意一下,机器人一次只能走一格,如果有些格子符合条件,但是如果它周边的点都不可达,那么该点也不可达。所以,一条路走到黑,不行就退一步,走其他的路。
网站上还提供一个广度优先搜索的方法,也是一种思路,两种方法具有相同复杂度
class Solution {
public:
int sums(int x){
int s = 0;
while(x != 0) {
s += x % 10;
x = x / 10;
}
return s;
}
int movingCountCore(int m,int n,int index_x,int index_y,bool* visited,int k){
int res = 0;
if(index_x >= 0 && index_x < n && index_y >= 0 && index_y < m ){
if((sums(index_x)+sums(index_y)) <= k && !visited[index_y*n + index_x]){
visited[index_y*n + index_x] = true;
res = 1 + movingCountCore(m,n,index_x+1,index_y,visited,k)
+ movingCountCore(m,n,index_x-1,index_y,visited,k)
+ movingCountCore(m,n,index_x,index_y+1,visited,k)
+ movingCountCore(m,n,index_x,index_y-1,visited,k);
}
}
return res;
}
int movingCount(int m, int n, int k) {
if(0 >= m || 0 >= n ) return 0;
bool *visited = (bool*)malloc(m*n*sizeof(bool));
memset(visited,0,m*n*sizeof(bool));
int index_x = 0;
int index_y = 0;
int res = movingCountCore(m, n, index_x, index_y, visited, k);
free(visited);
return res;
}
};