问题描述(原题链接)
地上有一个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。请问该机器人能够到达多少个格子?
代码:
深度优先搜索
class Solution {
int num=0;
public int movingCount(int m, int n, int k) {
boolean[][] visit = new boolean[m][n];
dfs(m,n,0,0,visit,k);
return num;
}
private int count(int line,int row){
int linecountmod=0;
int rowcountmod =0;
while(line !=0){
linecountmod+=line%10;
line = line/10;
}
while(row !=0){
rowcountmod+=row%10;
row = row/10;
}
return linecountmod+rowcountmod;
}
private void dfs(int m, int n,int i,int j, boolean[][] visit, int k){
if(i<0 || j<0 || i>=m || j>=n || visit[i][j] == true || count(i,j)>k){
return;
}
if(count(i,j)<=k)
num++;
visit[i][j]=true;
dfs(m,n,i+1,j,visit,k);
dfs(m,n,i-1,j,visit,k);
dfs(m,n,i,j+1,visit,k);
dfs(m,n,i,j-1,visit,k);
}
}
广度优先搜索
class Solution {
int num=0;
public int movingCount(int m, int n, int k) {
//广度优先搜索
boolean[][] visit = new boolean[m][n];
Queue<int[]> q = new LinkedList<int[]>();
q.offer(new int[]{0,0});
int[] dx = new int[]{0,1};
int[] dy = new int[]{1,0};
if(count(0,0)<=k){
num++;
visit[0][0]=true;
}
while(!q.isEmpty()){
int[] temp = q.poll();
int x = temp[0];
int y = temp[1];
for(int i=0;i<2;i++){
int newx = x+dx[i];
int newy = y +dy[i];
if(newx>=m || newy>=n || visit[newx][newy] || count(newx,newy)>k)
continue;
num++;
visit[newx][newy]=true;
q.offer(new int[]{newx,newy});
}
}
return num;
}
private int count(int line,int row){
int linecountmod=0;
int rowcountmod =0;
while(line !=0){
linecountmod+=line%10;
line = line/10;
}
while(row !=0){
rowcountmod+=row%10;
row = row/10;
}
return linecountmod+rowcountmod;
}
}