面试题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

示例 1:

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

class Solution {
    public int movingCount(int m, int n, int k) {
        int[] dx = {-1, 0, 1, 0};
        int[] dy = {0, -1, 0, 1};
        boolean[][] mark = new boolean[m][n];
        mark[0][0] = true;
        Queue<int[]> que = new LinkedList<>();
        que.add(new int[]{0, 0});
        while (!que.isEmpty()) {
            int[] cur = que.poll();
            for (int i = 0; i < 4; i++) {
                int x = cur[0] + dx[i];
                int y = cur[1] + dy[i];
                if (x >= 0 && y >= 0 && x < m && y < n && !mark[x][y] && !isOver(x, y, k)) {
                    mark[x][y] = true;
                    que.add(new int[]{x, y});
                }
            }
        }
        int count = 0;
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (mark[i][j]) {
                    count++;
                }
            }
        }
        return count;
    }

    private boolean isOver(int i, int j, int k) {
        int res = 0;
        while (i > 0) {
            res += i % 10;
            i /= 10;
        }
        while (j > 0) {
            res += j % 10;
            j /= 10;
        }
        return res > k;
    }
}

思路概述:
定义一个boolen数组,来记录机器人是否到达过该点,然后判断能到达的点位的上下左右四个位置是否可以达到。
注意:
不能单纯的通过遍历一个行为n、列为m的二维数组,通过计算数位来判断是否可以到达,因为有的数位满足要求,但是机器人并无法到达该点,因为前往该点的路上的点位无法到达。

发布了258 篇原创文章 · 获赞 5 · 访问量 5321

猜你喜欢

转载自blog.csdn.net/weixin_43105156/article/details/104734753