1.题目描述
2.解题思路
递推:
定义 vis[i][j] 为 (i, j) 坐标是否可达,如果可达返回 1,否则返回 0。
首先 (i, j) 本身需要可以进入,因此需要先判断 i 和 j 的数位之和是否大于 k ,如果大于的话直接设置 vis[i][j] 为不可达即可。
否则,前面提到搜索方向只需朝下或朝右,因此 (i, j) 的格子只会从 (i - 1, j) 或者 (i, j - 1) 两个格子走过来(不考虑边界条件),那么 vis[i][j] 是否可达的状态则可由如下公式计算得到:
vis[i][j]=vis[i−1][j] or vis[i][j−1]
即只要有一个格子可达,那么 (i, j) 这个格子就是可达的,因此我们只要遍历所有格子,递推计算出它们是否可达然后用变量 ans 记录可达的格子数量即可。
初始条件 vis[i][j] = 1 ,递推计算的过程中注意边界的处理。
复杂度分析
时间复杂度:O(mn),其中 m 为方格的行数, n 为方格的列数。一共有 O(mn) 个状态需要计算,每个状态递推计算的时间复杂度为 O(1),所以总时间复杂度为 O(mn)。
空间复杂度:O(mn),其中 m 为方格的行数,n 为方格的列数。我们需要 O(mn) 大小的结构来记录每个位置是否可达。
3.代码实现
class Solution {
public int movingCount(int m, int n, int k) {
if (k == 0) {
return 1;
}
boolean[][] vis = new boolean[m][n];
int ans = 1;
vis[0][0] = true;
for (int i = 0; i < m; ++i) {
for (int j = 0; j < n; ++j) {
if ((i == 0 && j == 0) || get(i) + get(j) > k) {
continue;
}
// 边界判断
if (i - 1 >= 0) {
vis[i][j] |= vis[i - 1][j];
}
if (j - 1 >= 0) {
vis[i][j] |= vis[i][j - 1];
}
ans += vis[i][j] ? 1 : 0;
}
}
return ans;
}
private int get(int x) {
int res = 0;
while (x != 0) {
res += x % 10;
x /= 10;
}
return res;
}
}