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

最新推荐文章于 2022-09-09 22:43:00 发布

HarvestWu

最新推荐文章于 2022-09-09 22:43:00 发布

阅读量283

点赞数

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.csdn.net/HarvestWu/article/details/109457491

版权

LeetCode 专栏收录该内容

458 篇文章 1 订阅

订阅专栏

目录结构

1.题目

2.题解

1.题目

地上有一个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。请问该机器人能够到达多少个格子？

示例：

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


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

2.题解

BFS

public class Offer13 {

    @Test
    public void test() {
        System.out.println(movingCount(20, 10, 9));
    }

    public int movingCount(int m, int n, int k) {
        if (k == 0) {
            return 1;
        }
        Queue<int[]> queue = new LinkedList<>();
        int[] dx = {0, 1}, dy = {1, 0};
        boolean[][] visited = new boolean[m][n];
        queue.offer(new int[]{0, 0});
        visited[0][0] = true;
        int count = 1;
        while (!queue.isEmpty()) {
            int[] position = queue.poll();
            int x = position[0], y = position[1];
            for (int i = 0; i < 2; i++) {
                int tx = x + dx[i], ty = y + dy[i];
                if (tx < m && ty < n && !visited[tx][ty] && getSum(tx) + getSum(ty) <= k) {
                    count++;
                    visited[tx][ty] = true;
                    queue.offer(new int[]{tx, ty});
                }
            }
        }
        return count;
    }

    public int getSum(int x) {
        int sum = 0;
        while (x != 0) {
            sum += x % 10;
            x /= 10;
        }
        return sum;
    }
}