2024-01-03 11:46:50 +0000 UTC

Number of Laser Beams in a Bank

Code

class Solution:
    def numberOfBeams(self, bank):
        ans, temp = 0, 0
        for s in bank:
            n = s.count('1')
            if n == 0:
                continue
            ans += temp * n
            temp = n
        return ans