地上有一个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 {
private int res = 0;
public int movingCount(int m, int n, int k) {
int [][] visit = new int[m][n];
visit[0][0] = 1;
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (check(i,j,k)){
if (i>0){//如果上面的格子为1,则说明这个可以从上面这个格子下来
visit[i][j] |= visit[i-1][j];
}
if (j>0){//使用或等于是防止在这个地方出现更改
visit[i][j] |= visit[i][j-1];
}
res+=visit[i][j];
}
}
}
return res;
}
private boolean check(int i, int j, int k) {
int count = 0;
int t = i;
while (t>0){
count+=t%10;
t = t/10;
}
t = j;
while (t>0){
count+=t%10;
t = t/10;
}
return count <= k;
}
}
dfs解法
class Solution {
private int res = 0;
private int m;
private int n;
private int k;
public int movingCount(int m, int n, int k) {
int [][] visit = new int[m][n];
this.m=m;
this.n = n;
this.k = k;
dfs(0,0,visit);
return res;
}
private void dfs(int i, int j, int[][] visit) {
if (i>=m || j>=n || visit[i][j]==1){//判断边界条件
return;
}
if (check(i,j,k)){
res++;
visit[i][j]=1;
dfs(i+1,j,visit);//不需要回溯
dfs(i,j+1,visit);
}
}
private boolean check(int i, int j, int k) {
int count = 0;
int t = i;
while (t>0){
count+=t%10;
t = t/10;
}
t = j;
while (t>0){
count+=t%10;
t = t/10;
}
return count <= k;
}
}