题目:输入矩阵的行数m和列数n,要求m和n每位数字之和不能大于k,问机器人最多能到达几个元素。
public int movingCount(int m, int n, int k)
思路:dfs+剪枝。类比 矩阵中的路径 来做。
从(0,0)开始搜索,只能i+1(往下),j+1(往右)。那(0,0)就是在左上角。因为只从(0,0)一个点开始搜索,所以被访问过的点就不用回溯到未访问的状态(像矩阵中的路径中要从多个点开始搜索,那就需要回溯,否则就会影响别的点)。
由于不需要回退,则i+1(往下),j+1(往右)就以足够,所以dfs只需要写两个。
剪枝(限制和判断条件):1.不能越界 2.m和n的每位之和不能大于k 3.访问过的不再继续计数(用visited[i][j])
注意:1.不用回退
2.return的时候要加1,因为到这里i,j这个元素已经满足条件。
3.(0,0)为左上角(因为从(0,0)开始只能i+1(向下),j+1(向右)),又不用回退,所以只需要两个方向。
4.每位之和可以写成一个小函数public int eachDigitSum(int x)。
public int movingCount(int m,int n,int k){
if(m < 0 || n < 0 || k < 0){
return -999;
}
boolean[][] visited = new boolean[m][n];
return dfs(0,0,visited,k);
}
private int dfs(int i, int j, boolean[][] visited, int k) {
//剪枝(限制和判断条件)
if(i < 0 || i >= visited.length || j < 0 || j >= visited[0].length || (eachDigitSum(i) + eachDigitSum(j) > k)
|| visited[i][j]){
return 0;
}
visited[i][j] = true;
//可以写成两个(i+1为向下,j+1为向右),因为起点(0,0)在左上角,又不用回退,所以已经足够
return 1 + dfs(i+1,j,visited,k) + dfs(i,j+1,visited,k) + dfs(i-1,j,visited,k)
+ dfs(i,j-1,visited,k);
}
private int eachDigitSum(int x) {
int sum = 0;
while(x != 0){
sum += x % 10;
x = x / 10;
}
return sum;
}