2125. 银行中的激光束数量 - 代码解析与逻辑图解(C++)

LeetCode 2125. 银行中的激光束数量 - C++代码解析与逻辑图解

题目描述

在一个银行的安全系统中，有一排安装了安全设备的银行柜台。每个柜台用一个二进制字符串表示，其中 ‘0’ 表示没有设备，‘1’ 表示有设备。激光束会在两个设备之间形成，当且仅当它们位于不同的柜台（行）且处于同一列，并且中间没有其他设备。给定一个表示银行柜台的字符串数组 bank，计算激光束的总数量。

解题思路

核心思路

激光束的形成需要满足两个条件：

• 两个设备位于不同的行（柜台）。- 两个设备位于同一列，且中间行的对应列没有设备。
  因此，解题的关键在于：

• 统计每行设备数量：遍历每一行，记录该行中设备（‘1’）的数量。- 计算相邻行激光束：对于每一对相邻的非零设备行，激光束数量为两行设备数量的乘积（每列设备一一对应形成光束）。

解题思路思维导图

代码解析

完整代码（带注释）

#include <vector>
#include <string>
#include <iostream>
using namespace std;

class Solution {
public:
    int numberOfBeams(vector<string>& bank) {
        int res = 0; // 激光束总数
        vector<int> dp(bank.size(), 0); // 存储每行设备数量
        vector<bool> hasBeam(bank.size(), false); // 标记该行是否有设备

        // 第一步：统计每行设备数量
        for (int i = 0; i < bank.size(); ++i) {
            for (int j = 0; j < bank[0].size(); ++j) {
                if (bank[i][j] == '1') {
                    dp[i]++; // 设备数量加1
                    hasBeam[i] = true; // 标记该行有设备
                }
            }
        }

        // 第二步：计算相邻设备行的激光束数量
        for (int i = 0; i < bank.size(); ++i) {
            // 遍历当前行之后的所有行
            for (int j = i + 1; j < bank.size() && hasBeam[i]; ++j) {
                if (hasBeam[j]) { // 找到下一个有设备的行
                    res += dp[i] * dp[j]; // 激光束数量 = 两行设备数乘积
                    i = j - 1; // 跳过中间行，避免重复计算
                    break; // 处理完当前行，跳出内层循环
                }
            }
        }

        return res;
    }
};

// 测试用例
void test01() {
    Solution s;
    vector<string> bank = { "011001","000000","010100","001000" };
    int res = s.numberOfBeams(bank);
    cout << "激光束数量: " << res << endl; // 输出：8
}

int main() {
    test01();
    system("pause");
    return 0;
}

代码步骤解析

1. 设备统计阶段

• 遍历每行每列：检查每个柜台（字符）是否为 ‘1’。- 记录设备数量：用 dp[i] 存储第 i 行的设备总数。- 标记有效行：用 hasBeam[i] 标记第 i 行是否有设备，避免后续无效遍历。

2. 激光束计算阶段

• 寻找相邻设备行：外层循环遍历每行，内层循环寻找下一个有设备的行。- 计算激光束数量：若找到下一个有设备的行 j，则激光束数量为 dp[i] * dp[j]。- 优化遍历：通过 i = j - 1 跳过中间无设备行，减少重复计算。

示例计算步骤

以测试用例 bank = {"011001","000000","010100","001000"} 为例，详细计算过程如下：

步骤1：统计每行设备数量

步骤2：计算激光束数量

激光束分布示意图

复杂度分析

时间复杂度

• 统计设备阶段：O(M*N)，其中 M 为行数，N 为列数，需遍历每个单元格。- 计算激光束阶段：O(M^2)，最坏情况下每行都有设备，需遍历所有行对。- 总体复杂度：O(MN + M^2)，当 M 较小时（如银行行数有限），接近 O(MN)。

空间复杂度

• 辅助数组：O(M)， dp 和 hasBeam 数组大小均为 M（行数）。- 总体复杂度：O(M)，与输入规模线性相关。

测试结果

功能验证

测试用例执行结果显示，激光束数量计算正确，代码通过所有测试点。

边界测试

题目链接

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

源码下载

https://wwbq.lanzn.com/ijGyc39gcu0b
密码:7usd