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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

示例 1:

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

示例 2:

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

提示:

1 <= n,m <= 100
0 <= k <= 20

解决方案

每个格子除本身数位之和要小于或等于k外,还需其上方或左侧可以被到达。

class Solution {
    public int movingCount(int m, int n, int k) {
        boolean[][] matrix = new boolean[m][n];
        int result = 0;
        matrix[0][0] = true;
        for (int row = 0; row < m; ++row) {
            for (int col = 0; col < n; ++col) {
                if (getDigitSum(row) + getDigitSum(col) > k) {
                    continue;
                }
                if (row - 1 >= 0) {
                    matrix[row][col] |= matrix[row - 1][col];
                }
                if (col - 1 >= 0) {
                    matrix[row][col] |= matrix[row][col - 1];
                }
                if (matrix[row][col]) {
                    result += 1;
                }
            }
        }
        return result;
    }

    private int getDigitSum(int number) {
        int sum = 0;
        while (number > 0) {
            sum += number % 10;
            number /= 10;
        }
        return sum;
    }
}

发表评论

您的电子邮箱地址不会被公开。 必填项已用*标注

此站点使用Akismet来减少垃圾评论。了解我们如何处理您的评论数据