机器人的运动范围

机器人的运动范围

题目描述

地上有一个 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:

1
2
输入:m = 2, n = 3, k = 1
输出:3

示例 2:

1
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
题目著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法一

使用广度优先遍历来访问二维矩阵,在访问的过程中判断路径和记录已经访问的点即可。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
class Node {
int x;
int y;

public Node(int x, int y) {
this.x = x;
this.y = y;
}
}

class Solution {
private int[][] directions = new int[][]{{0, 1}, {1, 0}, {0, -1}, {-1, 0}};

private static int add(int x, int y) {
int sum = 0;
while (x != 0) {
sum += x % 10;
x /= 10;
}
while (y != 0) {
sum += y % 10;
y /= 10;
}
return sum;
}

public int movingCount(int m, int n, int k) {
if (m == 0 || n == 0) return 0;
Node node = new Node(0, 0);
int count = 0;
boolean[][] visited = new boolean[m][n];
Queue<Node> queue = new ArrayDeque<>(m * n);
queue.add(node);
visited[0][0] = true;
while (!queue.isEmpty()) {
node = queue.poll();
count++;
int x = node.x;
int y = node.y;
for (int i = 0; i < 2; i++) {
int[] direction = directions[i];
int x2 = x + direction[0];
int y2 = y + direction[1];
if (x2 < 0 || y2 < 0 || x2 >= m || y2 >= n || visited[x2][y2] || add(x2, y2) > k) {
continue;
}
queue.add(new Node(x2, y2));
visited[x2][y2] = true;
}
}
return count;
}
}