地上有一个 m 行和 n 列的方格,横纵坐标范围分别是 0∼m−1 和 0∼n−1。
一个机器人从坐标0,0的格子开始移动,每一次只能向左,右,上,下四个方向移动一格。
但是不能进入行坐标和列坐标的数位之和大于 k 的格子。
请问该机器人能够达到多少个格子?
样例1
输入:k=7, m=4, n=5
输出:20
样例2
输入:k=18, m=40, n=40
输出:1484
解释:当k为18时,机器人能够进入方格(35,37),因为3+5+3+7 = 18。
但是,它不能进入方格(35,38),因为3+5+3+8 = 19。
注意:
0<=m<=50
0<=n<=50
0<=k<=100
注意
- JavaScript二维数组的初始化
- 始终要对特殊数组做预处理
- 这是一个典型的宽度优先搜索问题
- 步进条件:
- 满足数位之和小于k。
- 上侧或左侧已经被访问过了。
const k = 9, m = 41, n = 37;
const movingCount = function (threshold, rows, cols) {
if(!rows || !cols){return 0} // 处理特殊数组
let avaliableSum = 1;
let tempArr = Array(rows).fill(0); //ES5产生数组,ES6填充
let arr = [];
for (let k = 0; k < cols; k++) {
arr[k] = [...tempArr]; // 数组深拷贝,这里如果不使用数组解析,直接赋值,实际上只是把tempArr的地址传递出去了
}
arr[0][0] = 1; // 初始化机器人定位
for (let i = 0; i < cols; i++) { // 宽度优先搜索
for (let j = 0; j < rows; j++) {
// 可以向前步进的条件,1.满足数位之和小于k。 2.上侧或左侧已经被访问过了。
if ((getBitSum(i) + getBitSum(j) <= threshold) && ((i >= 1 && arr[i - 1][j] === 1) || (j >= 1 && arr[i][j - 1] === 1))) {
console.log(`[${i},${j}]`);
arr[i][j] = 1;
avaliableSum++;
}
}
}
console.log(arr);
return avaliableSum;
};
const getBitSum = (a) => {
let sum = 0;
while (Math.floor(a / 10)) {
sum += a % 10;
a = Math.floor(a / 10);
}
sum += a;
return sum;
}
console.log(movingCount(k, m, n));