地上有一个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; } }