剑指offer13. 机器人的运动范围
思路:
与剑指offer第十二题相似,用的是回溯法,首先判断该位置是否满足矩阵要求,是否满足题目要求(行坐标和列坐标的数位之和大于k的格子)以及该位置是否未被访问(设置一个boolean的矩阵来记录该位置是否被访问且被计数了),如果上述条件均满足就count加1并判断它的上下左右四个位置是否也满足条件。
代码:
class Solution {
public int movingCount(int m, int n, int k) {
//前提条件
if (m < 0 || n < 0 || k < 0)
return 0;
//记录是否已经访问,默认全为false,代表还未访问
boolean[] visit = new boolean[m * n];
//从[0,0]开始判断
return movingCountCore(m, n, 0, 0, k, visit);
}
public int movingCountCore(int m, int n, int row, int col, int k, boolean[] visit) {
//计数
int count = 0;
//该位置是否满足条件
if (cheak(m, n, row, col, k, visit)) {
visit[row * n + col] = true;
count = 1 + movingCountCore(m, n, row + 1, col, k, visit) + movingCountCore(m, n, row - 1, col, k, visit) + movingCountCore(m, n, row, col + 1, k, visit) + movingCountCore(m, n, row, col - 1, k, visit);
}
return count;
}
//判断是否满足题目要求
public boolean cheak(int m, int n, int row, int col, int k, boolean[] visit) {
if (row >= 0 && row < m && col >= 0 && col < n && !visit[row * n + col] && getDigitSum(row) + getDigitSum(col) <= k) {
return true;
}
return false;
}
//计算一个数的每位之和
public int getDigitSum(int number) {
int sum = 0;
while (number > 0) {
sum += (number % 10);
number /= 10;
}
return sum;
}
}