Time: 10 ms (54.97%), Space: 19.7 MB (25.73%) - LeetHub

Author: hogan-tech

2244-number-of-laser-beams-in-a-bank/2244-number-of-laser-beams-in-a-bank.py

@@ -0,0 +1,18 @@
+# time complexity: O(m * n)
+# space complexity: O(1)
+from typing import List
+
+
+class Solution:
+    def numberOfBeams(self, bank: List[str]) -> int:
+        prev = 0
+        laserSum = 0
+        for row in bank:
+            if row.count('1'):
+                laserSum += prev * row.count('1')
+                prev = row.count('1')
+        return laserSum
+
+
+bank = ["011001", "000000", "010100", "001000"]
+print(Solution().numberOfBeams(bank))