题目:
地上有一个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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
需要注意的地方:
- java 中队列的实现方法
- 求某个数每个位的数和的方法
- bfs 模板
题解:
class Solution {
Queue<int[]> q = new LinkedList<int[]>();
int res = 0;
public int movingCount(int m, int n, int k) {
bfs(m, n, k);
// 因为在bfs中起始点(0, 0)没有进行判断,又0 + 0 = 0,0 <= k <= 20,所以一定满足 <= k,所以 res 要 +1
return res + 1;
}
public void bfs(int m, int n, int k) {
int[][] dxy = {{-1, 0}, {0, -1}, {1, 0}, {0, 1}};
boolean[][] book = new boolean[m + 5][n + 5];
q.offer(new int[]{0, 0});
// 标记数组,防止对已经到达的格子,再次到达该格子
book[0][0] = true;
// bfs 模板的操作
while(!q.isEmpty()) {
int[] dir = q.poll();
for(int i = 0; i < 4; i ++) {
int x = dir[0] + dxy[i][0];
int y = dir[1] + dxy[i][1];
// 判断的条件
if(x >= 0 && y >= 0 && x < m && y < n && bitSum(x) + bitSum(y) <= k && book[x][y] == false) {
res ++;
q.offer(new int[]{x, y});
book[x][y] = true;
}
}
}
}
// 求某个数的每个位相加的和
public int bitSum(int num) {
int sum = 0;
while(num != 0) {
sum += num % 10;
num /= 10;
}
return sum;
}
}