题目描述:
地上有一个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
解题思路:
本题与 矩阵中的路径 类似,是典型的搜索 & 回溯问题。在介绍回溯算法算法前,为提升计算效率,首先讲述两项前置工作: 数位之和计算 、 可达解分析 。
数位之和计算:
设一数字 x ,向下取整除法符号 // ,求余符号 ⊙ ,则有:
x⊙10 :得到 x 的个位数字;
x//10 : 令 x 的十进制数向右移动一位,即删除个位数字。
因此,可通过循环求得数位和 s ,数位和计算的封装函数如下所示:
int sums(int x)
int s = 0;
while(x != 0) {
s += x % 10;
x = x / 10;
}
return s;
由于机器人每次只能移动一格(即只能从 x 运动至 x±1),因此每次只需计算 x 到 x±1 的数位和增量。本题说明 1≤n,m≤100,以下公式仅在此范围适用。
数位和增量公式: 设 x 的数位和为 s , x+1 的数位和为 s_{x+1} ;
当 (x+1)⊙10=0 时: s_{x+1} = s_x - 8 ,例如 19,20 的数位和分别为 10,2;
当 (x+1)⊙10≠0 时: s_{x+1} = s_x + 1 ,例如 1,2 的数位和分别为 1,2 。
以下代码为增量公式的三元表达式写法,将整合入最终代码中。
(x + 1) % 10 != 0 ? s_x + 1 : s_x - 8;
根据可达解的结构和连通性,易推出机器人可 仅通过向右和向下移动,访问所有可达解 。
深度优先遍历 DFS
深度优先搜索: 可以理解为暴力法模拟机器人在矩阵中的所有路径。DFS 通过递归,先朝一个方向搜到底,再回溯至上个节点,沿另一个方向搜索,以此类推。
剪枝: 在搜索中,遇到数位和超出目标值、此元素已访问,则应立即返回,称之为 可行性剪枝 。
Java程序:
class Solution {
int m, n, k;
boolean[][] visited;
public int movingCount(int m, int n, int k) {
this.m = m; this.n = n; this.k = k;
this.visited = new boolean[m][n];
return dfs(0, 0, 0, 0);
}
public int dfs(int i, int j, int si, int sj) {
if(i >= m || j >= n || k < si + sj || visited[i][j]) return 0;
visited[i][j] = true;
return 1 + dfs(i + 1, j, (i + 1) % 10 != 0 ? si + 1 : si - 8, sj) + dfs(i, j + 1, si, (j + 1) % 10 != 0 ? sj + 1 : sj - 8);
}
}