[路飞]_leetcode274 周赛2

186 阅读1分钟

5968. 银行中的激光束数量

题目

银行内部的防盗安全装置已经激活。给你一个下标从 0 开始的二进制字符串数组 bank ,表示银行的平面图,这是一个大小为 m x n 的二维矩阵。 bank[i] 表示第 i 行的设备分布,由若干 '0' 和若干 '1' 组成。'0' 表示单元格是空的,而 '1' 表示单元格有一个安全设备。

对任意两个安全设备而言,如果同时 满足下面两个条件,则二者之间存在 一个 激光束:

两个设备位于两个 不同行 :r1 和 r2 ,其中 r1 < r2 。 满足 r1 < i < r2 的 所有 行 i ,都 没有安全设备 。 激光束是独立的,也就是说,一个激光束既不会干扰另一个激光束,也不会与另一个激光束合并成一束。

返回银行中激光束的总数量。

示例1

image.png

输入:bank = ["011001","000000","010100","001000"]
输出:8
解释:在下面每组设备对之间,存在一条激光束。总共是 8 条激光束:
 * bank[0][1] -- bank[2][1]
 * bank[0][1] -- bank[2][3]
 * bank[0][2] -- bank[2][1]
 * bank[0][2] -- bank[2][3]
 * bank[0][5] -- bank[2][1]
 * bank[0][5] -- bank[2][3]
 * bank[2][1] -- bank[3][2]
 * bank[2][3] -- bank[3][2]
注意,第 0 行和第 3 行上的设备之间不存在激光束。
这是因为第 2 行存在安全设备,这不满足第 2 个条件。

题解

这题也比较简单

  • 排除全是0的列
  • 如果只有1列返回0;
  • 将相邻两列的字符串为1的数量找到并且向乘
  • 得到结果 代码如下:

代码

var numberOfBeams = function (bank) {
  const list = bank.filter((v) => Number(v) > 0)

  const len = list.length
  if (len === 1) return 0
  let result = 0
  for (let i = 0; i < len - 1; i++) {
    const current = getNum(list[i])
    const next = getNum(list[i + 1])
    result += current * next
  }
  return result

  function getNum(s) {
    let num = 0
    for (let i = 0; i < s.length; i++) {
      if (s[i] === '1') num++
    }
    return num
  }
}