题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
地上有一个m行n列的方格,从坐标 [0,0] 到坐标 [m-1,n-1] 。一个机器人从坐标 [0, 0] 的格子开始移动,它每次可以向左、右、上、下移动一格(不能移动到方格外),也不能进入行坐标和列坐标的数位之和大于k的格子。例如,当k为18时,机器人能够进入方格 [35, 37] ,因为3+5+3+7=18。但它不能进入方格 [35, 38],因为3+5+3+8=19。请问该机器人能够到达多少个格子?
示例 1:
输入:m = 2, n = 3, k = 1
输出:3
示例 2:
输入:m = 3, n = 1, k = 0
输出:1
可以利用广度优先搜索遍历,从最左上角的位置开始,每次都向右边和下边进行探索,并判断能否符合题目要求,如果符合,就把计数器增加。
int counts = 0; //记录能走的格子数
public int movingCount(int m, int n, int k) {
int[][] visit = new int[m][n]; //创建辅助数组,标记走过了的格子
isVisit(visit,0,0,m-1,n-1,k);
return counts;
}
/**
* 广度优先搜索遍历,把没访问过且符合题意的位置记为1,
* 并向右和下进行下一次访问
* @param visit
* @param i 横坐标
* @param j 纵坐标
* @param m 横边界
* @param n 纵边界
* @param k 题目限制的数字k
*/
public void isVisit(int[][] visit,int i,int j,int m,int n,int k){
if (i>=0&&j>=0&&i<=m&&j<=n&&visit[i][j]==0&&(idxSum(i)+idxSum(j))<=k) {
counts++;
visit[i][j]=1;
//isVisit(visit, i - 1, j, m, n, k); //完全可以只向右或者向下搜索,因为左上方向的肯定已经搜索过了
isVisit(visit, i + 1, j, m, n, k);
//isVisit(visit, i, j - 1, m, n, k);
isVisit(visit, i, j + 1, m, n, k);
}
}
/**
* 获取整数的数位之和
* @param x
* @return 数位之和 sum
*/
public int idxSum(int x){
int sum = x%10;
int temp = x/10;
while (temp>0){
sum+=temp%10;
temp/=10;
}
return sum;
}