题目:
链接:剑指 Offer 13. 机器人的运动范围
难度:中等
地上有一个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。请问该机器人能够到达多少个格子?
示例 1:
输入:m = 2, n = 3, k = 1
输出:3
示例 2:
输入:m = 3, n = 1, k = 0
输出:1
提示:
- 1 <= n,m <= 100
- 0 <= k <= 20
深度优先搜索:
简单的深搜。从坐标原点(0,0)开始不断向下或向右搜索(从左上角开始搜索,所以不用向上或向左就可以全部遍历完),用一个visited数组标记每个格子是否已到达,搜索过程中判断是否越界/已到达/数位之和不大于k,符合条件的统计+1。
详细请看代码注释。
代码:
class Solution {
public:
int ans = 0; //从全局统计能到达的格子数量
int m, n, k;
vector<vector<bool>> visited; // 标记每个格子是否已到达
int digitCount(int a, int b) { // 统计两个整数的数位之和
int num = 0;
while(a || b) {
num += (a % 10);
num += (b % 10);
a /= 10;
b /= 10;
}
return num;
}
void dfs(int i, int j) {
if(i < 0 || i >= m || j < 0 || j >= n) return; // 边界判断
if(visited[i][j]) return; // 格子已经被到达过
if(digitCount(i , j) > k) return; // 坐标数位之和 > k,不能到达
ans++; // 可以到达,统计+1
visited[i][j] = true; // 标记已到达
dfs(i + 1, j); // 从当前格子向下或向右步进
dfs(i, j + 1);
}
int movingCount(int m, int n, int k) {
this->m = m;
this->n = n;
this->k = k;
visited.resize(m, vector<bool>(n, false)); // 初始化访问数组
dfs(0, 0); // 从坐标原点开始搜索
return ans;
}
};
时间复杂度:O(M * N),因为每个格子最多只会被访问两次(从左边格子或上边格子步进)。
空间复杂度:O(M * N),visited 标记数组的大小是M * N。