2125. 银行中的激光束数量

leetcode力扣刷题打卡

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

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

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

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

原代码##

class Solution {
public:
    int numberOfBeams(vector<string>& bank) {
        vector<pair<int, int>> v;
        for (int i = 0; i < bank.size(); ++i) {
            int cnt = 0;
            for (int j = 0; j < bank[i].size(); ++j) {
                if (bank[i][j] == '1') {
                    cnt++;
                }
            }
            if (cnt != 0)
                v.emplace_back(i, cnt);
        }
        if(v.empty()) return 0;
        int sum = 0;
        for (int i = 0; i < v.size() - 1; ++i) {
            sum += v[i].second * v[i + 1].second;
        }
        return sum;
    }
};

思路：

1、用一个二维数组记录行号和1的数量;
2、可能出现v为空的情况，所以要先进行判断。