【leetcode】剑指Offer面试题11 . 12
面试题13. 机器人的运动范围
地上有一个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
输入:m = 3, n = 1, k = 0
输出:1
提示:1 <= n,m <= 100
0 <= k <= 20
思路:这个题是很典型的矩阵问题,但是一般的矩阵可达问题都是有明显障碍物分布在矩阵中,而该题的障碍物是隐性的,即:坐标数位之和大于k;这就需要我们分清什么坐标是真正的障碍,什么坐标是被障碍阻挡得不可达目标。这个问题绘图会很容易得到:
通过观察可得,若k为9,橙色区域为障碍,浅绿区域为可达,墨绿区域为不可达。并且可达区域会随着k值的变化而变化。
对于这类问题,通常使用深度优先搜索或广度优先搜索来解:
深度优先搜索(DFS):
class Solution {
int m, n, k;
boolean[][] visited;
public int movingCount(int m, int n, int k) {
this.m = m; this.n = n; this.k = k;
this.visited = new boolean[m][n];
return dfs(0, 0, 0, 0);
}
//ix、iy是数位和
public int dfs(int i, int j, int ix, int iy) {
if(i >= m || j >= n || k < ix + iy || visited[i][j]) return 0;
visited[i][j] = true;
return 1 + dfs(i + 1, j, (i + 1) % 10 != 0 ? ix + 1 : ix - 8, iy) + dfs(i, j + 1, ix, (j + 1) % 10 != 0 ? iy + 1 : iy - 8);
}
}
广度优先搜索(BFS)
class Solution {
public int movingCount(int m, int n, int k) {
boolean[][] visited = new boolean[m][n];
int res = 0;
Queue<int[]> queue= new LinkedList<int[]>();
queue.add(new int[] { 0, 0, 0, 0 });
while(queue.size() > 0) {
int[] x = queue.poll();
int i = x[0], j = x[1], ix = x[2], iy= x[3];
if(i >= m || j >= n || k < ix+ iy|| visited[i][j]) continue;
visited[i][j] = true;
res ++;
queue.add(new int[] { i + 1, j, (i + 1) % 10 != 0 ? ix+ 1 : ix- 8, iy});
queue.add(new int[] { i, j + 1, ix, (j + 1) % 10 != 0 ? iy + 1 : iy - 8 });
}
return res;
}
}