地上有一个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
思路:从坐标[0,0]开始深度优先搜索,并标识走过的格子。
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);
}
public int dfs(int i, int j){
//网格出界、格子已访问、格子坐标大于目标值
if(i < 0 || j < 0 || i >= m || j >= n || visited[i][j] || sum(i)+sum(j) > k){
return 0;
}
//访问标识
visited[i][j] = true;
//下右递归
return dfs(i+1, j) + dfs(i, j+1) + 1;
}
//数字各位之和
public int sum(int num){
int sum = 0;
while(num != 0){
sum = sum + num%10;
num = num / 10;
}
return sum;
}
值得注意的是,由于机器人是[0,0]作为起点往上下左右运动,而网格部分网格满足小于目标值的条件,但却是不可达的,即该区域是封闭的,被不满足条件的网格所包围。所以如果直接遍历网格每个格子进行计算得到的格子数是错误的。
广度优先搜索(效率更低)
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];
int count = 0;
Queue<int[]> queue = new LinkedList();
//起点入队
queue.add(new int[]{0, 0});
while(!queue.isEmpty()){
//出队
int[] temp = queue.poll();
int row = temp[0];
int col = temp[1];
if(row < 0 || row >= m || col < 0 || col >=n || visited[row][col] ||
sum(row) + sum(col) > k){
continue;
}
visited[row][col] = true;
count++;
//下右入队
queue.add(new int[]{row+1, col});
queue.add(new int[]{row, col+1});
}
return count;
}
参考:https://leetcode-cn.com/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof