2125. Number of Laser Beams in a Bank

[흑화뉴비(coderhs)]

왜 미디엄인지는 모르겠으나 시키는대로 하면 나온다.

class Solution:

    def numberOfBeams(self, bank: List[str]) -> int:

        ans = 0

        prev = 0

        for bb in bank:

            x = bb.count("1")

            if x == 0:

                continue

            if prev == 0:

                prev = x

            else:

                ans += prev * x

                prev = x

        return ans