机器人的运动范围

发表于 更新于 分类于 ， 阅读量 loading...

机器人的运动范围

题目描述

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