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

Eran_2023-10-08 16:23

题目

地上有一个 rows 行和 cols 列的方格。坐标从 [0,0] 到 [rows-1,cols-1] 。一个机器人从坐标 [0,0] 的格子开始移动，每一次只能向左，右，上，下四个方向移动一格，但是不能进入行坐标和列坐标的数位之和大于 threshold 的格子。例如，当 threshold 为 18 时，机器人能够进入方格 [35,37] ，因为 3+5+3+7 = 18。但是，它不能进入方格 [35,38] ，因为 3+5+3+8 = 19 。请问该机器人能够达到多少个格子？

数据范围：0≤threshold≤15 ，1≤rows,cols≤100

进阶：空间复杂度 O(nm) ，时间复杂度 O(nm)

示例1

输入：1,2,3

返回值：3

示例2

输入：0,1,3

返回值：1

示例3

输入：10,1,100

返回值：29

说明：[0,0],[0,1],[0,2],[0,3],[0,4],[0,5],[0,6],[0,7],[0,8],[0,9],[0,10],[0,11],[0,12],[0,13],[0,14],[0,15],[0,16],[0,17],[0,18],[0,19],[0,20],[0,21],[0,22],[0,23],[0,24],[0,25],[0,26],[0,27],[0,28] 这29种，后面的[0,29],[0,30]以及[0,31]等等是无法到达的

示例4

输入：5,10,10

返回值：21

解答

源代码

java 复制代码

import java.util.*;

public class Solution {
    public int movingCount(int threshold, int rows, int cols) {
        boolean[][] flag = new boolean[rows][cols];
        return dfs(threshold, rows, cols, 0, 0, flag);
    }

    public int dfs(int threshold, int rows, int cols, int i, int j, boolean[][] flag) {
        if (i < 0 || i >= rows || j < 0 || j >= cols || flag[i][j] || cal(i) + cal(j) > threshold) {
            return 0;
        }

        flag[i][j] = true;
        return 1 + dfs(threshold, rows, cols, i - 1, j, flag)
                        + dfs(threshold, rows, cols, i + 1, j, flag)
                        + dfs(threshold, rows, cols, i, j - 1, flag)
                        + dfs(threshold, rows, cols, i, j + 1, flag);
    }

    public int cal(int number) {
        int sum = 0;
        while (number != 0) {
            sum += number % 10;
            number /= 10;
        }

        return sum;
    }
}

总结

迷宫问题，回溯就完事，记得做标记避免重复经过方格。

这里要注意求的是机器人的运动"范围"而不是一条最大路径，所以每次应该加上上下左右路径的所有长度，而不是只加上一条最长的路径。