Merge pull request #10187 from simongreen-net/master

sgreen solutions to challenge 271

1 files changed, 47 insertions, 0 deletions

diff --git a/challenge-271/sgreen/python/ch-1.py b/challenge-271/sgreen/python/ch-1.py
new file mode 100755
index 0000000000..21f9744e1a
--- /dev/null
+++ b/challenge-271/sgreen/python/ch-1.py
@@ -0,0 +1,47 @@
+#!/usr/bin/env python3
+
+import json
+import sys
+
+
+def maximum_ones(matrix: list) -> int:
+    """
+    Finds the row with the maximum number of ones in a binary matrix.
+
+    Args:
+        matrix (list): A binary matrix represented as a list of lists.
+
+    Returns:
+        int: The 1-based index of the row with the maximum number of ones.
+    """
+
+    rows = len(matrix)
+    cols = len(matrix[0])
+
+    # The row that has the maximum values
+    max_row = 0
+
+    # The number of ones in that row
+    max_count = 0
+
+    for row in range(rows):
+        # Check this has the same number of columns as the first row
+        if len(matrix[row]) != cols:
+            raise ValueError("Row %s has the wrong number of columns", row)
+
+        if sum(matrix[row]) > max_count:
+            max_row = row
+            max_count = sum(matrix[row])
+
+    # Return the rows as a 1-based index value
+    return max_row + 1
+
+
+def main():
+    matrix = json.loads(sys.argv[1])
+    result = maximum_ones(matrix)
+    print(result)
+
+
+if __name__ == '__main__':
+    main()