算法记录
LeetCode 题目:
地上有一个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。请问该机器人能够到达多少个格子?
说明
一、题目
地上有一个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暴力搜,前序判断是否能继续递归,走过的就计数器+1,并且标记走过。
class Solution {
int result = 0;
boolean[][] flag = null;
public int movingCount(int m, int n, int k) {
flag = new boolean[m][n];
DFS(m, n, k, 0, 0);
return result;
}
void DFS (int m, int n, int k, int x, int y) {
// 递归终止条件
int sum = 0;
int tx = x;
int ty = y;
while (tx > 0) {
sum += tx % 10;
tx /= 10;
}
while (ty > 0) {
sum += ty % 10;
ty /= 10;
}
if (x < 0 || y < 0 || x > m-1 || y > n - 1 || sum > k || flag[x][y]) {
return;
}
result++;
flag[x][y] = true;
int[][] four = {{x+1, y}, {x-1, y}, {x, y-1},{x, y+1}};
for (int i = 0; i < four.length; i++) {
int nx = four[i][0], ny = four[i][1];
DFS(m, n, k, nx, ny);
}
}
}
总结
做这DFS的题目最好在大脑把代码过程走一遍。