机器人的运动范围
题目:
地上有一个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
思路:
运用广度有优先搜索。设定队列保存待处理的数。设定一个数组保存已经处理过的数,达到除重的目的。
代码:
import java.util.LinkedList;
import java.util.Queue;
class Solution {
public int movingCount(int m, int n, int k) {
if(k==0){
return 1;
}
int res=1;
Queue<int[]> queue=new LinkedList<>();
queue.offer(new int[]{0,0});
int [][] nums=new int[m][n];
while(!queue.isEmpty()){
int[] cell=queue.poll();
int y=cell[0],x=cell[1];
if(y+1<m){
if(getNum(y+1)+getNum(x)<=k&&nums[y+1][x]!=1){
queue.offer(new int[]{y+1,x});
nums[y+1][x]=1;
res++;
}
}
if(x+1<n){
if(getNum(x+1)+getNum(y)<=k&&nums[y][x+1]!=1){
queue.offer(new int[]{y,x+1});
nums[y][x+1]=1;
res++;
}
}
}
return res;
}
public int getNum(int num) {
int res=0;
while (num!=0){
res+=num%10;
num=num/10;
}
return res;
}
}
浙公网安备 33010602011771号