剑指 Offer 13. 机器人的运动范围

剑指 Offer 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。请问该机器人能够到达多少个格子？

示例1：

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

示例2：

输入：m = 3, n = 1, k = 0
输出：1

数据限制：

1 <= n,m <= 100

0 <= k <= 20

思路：BFS

有题目的限制可知，我们只能从原点开始，走出一片连通的区域，这片连通的区域要符合两个条件：

• 处于方格的范围之内

• 设区域中某点的坐标为(x, y)，则x与y的数位之后要小于等于k

因此，我们可以从起点开始，走一遍BFS，所能经过的点，就是题目要求的点，之后返回走过的格子数即可

时间复杂度：O(nm)，最坏的情况下，所有的点都要遍历到

空间复杂度：O(nm)

代码：

class Solution {
    public int movingCount(int m, int n, int k) {
        int[][] dir = new int[][]{{0, 1}, {0, -1}, {1, 0}, {-1, 0}}; // 方向
        int[][] h = new int[10005][2];  
        boolean[][] map = new boolean[105][105];
        for (int i = 0; i < 100; i++){  // 初始化等
            Arrays.fill(map[i], true);
        }
        map[0][0] = false;
        int l = 0, r = 1;
        while(l < r){ // BFS
            int x = h[l][0];
            int y = h[l][1];
            for (int i = 0; i < 4; i++){
                int dx = x + dir[i][0];
                int dy = y + dir[i][1];
                if (dx < m && dy < n && dx >= 0 && dy >= 0 && map[dx][dy] && su(dx) + su(dy) <= k){
                    map[dx][dy] = false;
                    h[r][0] = dx;
                    h[r][1] = dy;
                    r++;
                }
            }
            l++;
        }
        return l; // 返回走过的格子数
    }
​
    int su(int x){  // 返回一个整数的数位之和
        int num = 0;
        while (x > 0){
            num += x % 10;
            x /= 10;
        }
        return num;
    }
}