地上有一个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。请问该机器人能够到达多少个格子?
输入:m = 2, n = 3, k = 1
输出:3
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof
- 一般遇到需要遍历整个矩阵找到答案的时候,可以考虑使用深度优先搜索或者广度优先搜索;
- 首先我们使用深度优先搜索,这里需要注意的是,我们既然要统计可以到达的格子数,那么一定要对已经遍历过的格子做个标记,不然后面再来搜索的时候会重复计算;这里有个重要的地方就是我们一开始就设置为从(0,0)开始走,所以我们只需要计算向右和向下的部分,不需要i-1和j-1;
class Solution {
int m;
int n;
int k;
int sum;
boolean[][] visited;
public int movingCount(int m, int n, int k) {
this.m = m;
this.n = n;
this.k = k;
visited = new boolean[m][n];
dfs(0,0);
return sum;
}
public void dfs(int i,int j){
if(i < 0 || j < 0 || i >= m || j >= n || visited[i][j] == true){
return;
}
int count = 0;
int im = i;
int jn = j;
while(im > 0){
count += (im % 10);
im /= 10;
}
while(jn > 0){
count += (jn % 10);
jn /= 10;
}
if(count > k){
return;
}
visited[i][j] = true;
sum++;
dfs(i+1,j);
dfs(i,j+1);
}
}
- 以下是广度优先搜索,我们一般使用队列去实现广度优先搜索,
class Solution {
public int movingCount(int m, int n, int k) {
Deque<int[]> queue = new LinkedList<>();
boolean[][] visited = new boolean[m][n];
queue.add(new int[]{0,0});
int count = 0;
while(queue.size() > 0){
int sum = 0;
int[] arr = queue.poll();
int i = arr[0];
int j = arr[1];
int im = i;
int jn = j;
if(i < 0 || j < 0 || i >= m || j >= n || visited[i][j] == true){
continue;
}
while(im > 0){
sum += (im % 10);
im /= 10;
}
while(jn > 0){
sum += (jn % 10);
jn /= 10;
}
if(sum > k){
continue;
}
visited[i][j] = true;
count++;
queue.add(new int[]{i+1,j});
queue.add(new int[]{i,j+1});
}
return count;
}
}