【路飞】算法与数据结构-剑指 Offer 13. 机器人的运动范围

109 阅读1分钟

不管全世界所有人怎么说,我都认为自己的感受才是正确的。无论别人怎么看,我绝不打乱自己的节奏。喜欢的事自然可以坚持,不喜欢的怎么也长久不了。

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。请问该机器人能够到达多少个格子?

示例 1:

输入: m = 2, n = 3, k = 1
输出: 3

示例 2:

输入: m = 3, n = 1, k = 0
输出: 1

思路

从起点开始扩散搜索周围的点, 当一个点满足条件后继续向周围搜索。

var movingCount = function (m, n, k) {
  var visted = new Array(m).fill(0).map(() => new Array(n).fill(false));
  var res = 0;
  var dfs = function (i, j) {
    //边界情况 && 确定是否已经遍历过
    if (i < 0 || j < 0 || i >= m || j >= n || visted[i][j]) return;
    visted[i][j] = true;
    // 开始查找
    if (canEnter(i, j, k)) {
      res++;
      dfs(i + 1, j);
      dfs(i, j + 1);
      dfs(i - 1, j);
      dfs(i, j - 1);
    }
  };
  dfs(0, 0);
  return res;
};

// 是否可以进入
var canEnter = function (m, n, k) {
  var res = 0;
  while (m) {
    res += m % 10;
    m = Math.floor(m / 10);
  }

  while (n) {
    res += n % 10;
    n = Math.floor(n / 10);
  }
  return res <= k;
};