LeetCode 2125.银行中的激光束数量

吃着火锅x唱着歌2024-03-02 22:07

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

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

两个设备位于两个不同行：r1 和 r2 ，其中 r1 < r2 。

满足 r1 < i < r2 的所有行 i ，都没有安全设备。

激光束是独立的，也就是说，一个激光束既不会干扰另一个激光束，也不会与另一个激光束合并成一束。

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

示例 1：

输入：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 个条件。
示例 2：

输入：bank = $"000","111","000"$

输出：0

解释：不存在两个位于不同行的设备

提示：

m == bank.length

n == bank $i$ .length

1 <= m, n <= 500

bank $i$ $j$ 为 '0' 或 '1'

直接模拟即可：

cpp 复制代码

class Solution {
public:
    int numberOfBeams(vector<string>& bank) {
        int ans = 0;
        int last = 0;
        for (string &row : bank)
        {
            int deviceNum = count(row.begin(), row.end(), '1');
            if (deviceNum == 0)
            {
                continue;
            }
            ans += deviceNum * last;
            last = deviceNum;
        }

        return ans;
    }
};