剑指offer—13.机器人的运动范围——分析及代码(Java)

一、题目

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

二、分析及代码

1. 回溯法

（1）思路

从原点开始，递归遍历所到位置各个方向所有格子及其附近格子的可达情况，并进行记录。

（2）代码

public class Solution {
    public int movingCount(int threshold, int rows, int cols)
    {
        boolean[][] visited = new boolean[rows][cols];
        for (int i = 0; i < rows; i++)
            for (int j = 0; j < cols; j++)
                visited[i][j] = false;
        int ans = 0;
        ans += canMoving(threshold, rows, cols, 0, 0, visited);
        return ans;
    }
    
    public int canMoving(int threshold, int rows, int cols, int i, int j, boolean[][] visited) {
        if (i >= 0 && i < rows && j >= 0 && j < cols && !visited[i][j] && sumLessThanThreshold(threshold, i, j)) {
            int sum = 1;
            visited[i][j] = true;
            sum += canMoving(threshold, rows, cols, i - 1, j, visited);
            sum += canMoving(threshold, rows, cols, i + 1, j, visited);
            sum += canMoving(threshold, rows, cols, i, j - 1, visited);
            sum += canMoving(threshold, rows, cols, i, j + 1, visited);
            return sum;
        }
        return 0;
    }
    
    boolean sumLessThanThreshold(int threshold, int i, int j) {
        int sum = 0;
        while (i > 0) {
            sum += i % 10;
            i /= 10;
        }
        while (j > 0) {
            sum += j % 10;
            j /= 10;
        }
        return (sum <= threshold);
    }
}

（3）结果

运行时间：17ms，占用内存：9364k。

三、其他

暂无。