机器人的运动范围
地上有一个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。请问该机器人能够到达多少个格子?
做法:典型的DFS题目
class Solution {
public int movingCount(int m, int n, int k) {
//已经走过的条件
boolean[][] flag = new boolean[m][n];
return dfs(m,n,0,0,k,flag);
}
public int dfs(int m,int n,int i,int j,int k,boolean[][] flag){
//边界条件
if(i < 0 || i >= m || j < 0 || j >= n || getSum(i,j) > k || flag[i][j]){
return 0;
}
//可以进入当前格子
flag[i][j] = true;
return dfs(m,n,i+1,j,k,flag)
+ dfs(m,n,i-1,j,k,flag)
+ dfs(m,n,i,j+1,k,flag)
+ dfs(m,n,i,j-1,k,flag)
+ 1;
}
public int getSum(int i,int j){
int sum = 0;
while(i != 0){
int digit = i % 10;
sum += digit;
i /= 10;
}
while(j != 0){
int digit = j % 10;
sum += digit;
j /= 10;
}
return sum;
}
}
自己遇到的一个编码臭习惯:
public int dfs(int m,int n,int i,int j,int k,boolean[][] flag){
//边界条件
if(i < 0 || i >= m || j < 0 || j >= n || flag[i][j]){
return 0;
}
//获取数位和
int sumI = 0;
while(i != 0){
int digit = i % 10;
sumI += digit;
i /= 10;
}
int sumJ = 0;
while(j != 0){
int digit = j % 10;
sumJ += digit;
j /= 10;
}
int sum = sumI + sumJ;
if(sum <= k){
//可以进入当前格子
flag[i][j] = true;
return dfs(m,n,i+1,j,k,flag)
+ dfs(m,n,i-1,j,k,flag)
+ dfs(m,n,i,j+1,k,flag)
+ dfs(m,n,i,j-1,k,flag) + 1;
}else{
return 0;
}
}
每次进dfs方法,在方法内运算sum,都将i和j置为0,导致StackOverFlow。