#2125

Number of Laser Beams in a Bank

Medium
ArrayMathStringMatrixHash MapArray
LeetCode ↗

Approaches

Brute ForceOptimal
Complexity Comparison
Brute ForceOptimal Solution
Time
O(n²)
O(n)
Space
O(1)
O(1)
💡

Intuition

Time O(n)Space O(1)

The optimal approach counts the number of devices in each row and only considers rows with devices. It uses a single pass to calculate the total beams by keeping track of the last row with devices and the count of devices in the current row.

⚙️

Algorithm

4 steps
  1. 1Step 1: Initialize a variable to count the total beams and a variable to store the count of devices in the last row with devices.
  2. 2Step 2: Loop through each row of the bank and count the devices in the current row.
  3. 3Step 3: If the current row has devices, multiply the count of devices in the last row with devices by the count in the current row and add to total beams.
  4. 4Step 4: Update the last row device count to the current row's device count.
solution.py9 lines
1def numberOfBeams(bank):
2    total_beams = 0
3    last_count = 0
4    for row in bank:
5        current_count = row.count('1')
6        if current_count > 0:
7            total_beams += last_count * current_count
8            last_count = current_count
9    return total_beams

Complexity note: The time complexity is O(n) because we only loop through the rows once, counting devices in each. The space complexity is O(1) since we only use a few variables for counting.

  • 1Laser beams can only exist between rows that have security devices, with no devices in between.
  • 2The number of beams between two rows is the product of the number of devices in each row.

Solutions and explanations are original Tejav content. Problem titles © LeetCode — use the LeetCode button above for the full problem statement.