aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rwxr-xr-x2610/2125/main.py26
1 files changed, 26 insertions, 0 deletions
diff --git a/2610/2125/main.py b/2610/2125/main.py
new file mode 100755
index 0000000..da45c29
--- /dev/null
+++ b/2610/2125/main.py
@@ -0,0 +1,26 @@
+#!/usr/bin/env python
+
+import typing
+
+
+class Solution:
+ def numberOfBeams(self, bank: typing.List[str]) -> int:
+ row_count = 0
+ result = 0
+ for row in bank:
+ count = row.count("1")
+ if count != 0:
+ result += row_count * count
+ row_count = row.count("1")
+
+ return result
+
+
+def main():
+ solution = Solution()
+ print(solution.numberOfBeams(["011001", "000000", "010100", "00100"]))
+ print(solution.numberOfBeams(["000", "111", "000"]))
+
+
+if __name__ == "__main__":
+ main()