机器人的运动范围

Posted 2020-08-12 码上哈希

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

回溯法：

public class Solution {
    public int movingCount(int threshold, int rows, int cols) {
        if (threshold == 0)
            return 1;
        if (threshold < 0)
            return 0;
        int[][] flag = new int[rows][cols];
        return move(threshold, rows, cols, flag, 0, 0);
    }
    
    public int move(int threshold, int rows, int cols, int[][] flag, int i, int j) {
        if (i >= rows || i < 0 || j >= cols || j < 0 || flag[i][j]==1 || bitSum(i)+bitSum(j)>threshold) {
            return 0;
        }
        flag[i][j] = 1;
        return move(threshold, rows, cols, flag, i+1, j)
            + move(threshold, rows, cols, flag, i-1, j)
            + move(threshold, rows, cols, flag, i, j+1)
            + move(threshold, rows, cols, flag, i, j-1)
            + 1;
    }
    
    public int bitSum(int n) {
        if (n == 0)
            return 0;
        int sum = 0;
        while (n != 0) {
            sum += n % 10;
            n /= 10;
        }
        return sum;
    }
}