Merge pull request #10187 from simongreen-net/master

sgreen solutions to challenge 271

3 files changed, 102 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()
diff --git a/challenge-271/sgreen/python/ch-2.py b/challenge-271/sgreen/python/ch-2.py
new file mode 100755
index 0000000000..7ebb7f412a
--- /dev/null
+++ b/challenge-271/sgreen/python/ch-2.py
@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+
+import sys
+
+
+def sort_by_1_bits(ints: list) -> list:
+    """
+    Sorts a list of integers based on the number of set bits (1s) in their
+    binary representation. If they are equal, they are sorted by their value.
+
+    Args:
+        ints (list): A list of integers to be sorted.
+
+    Returns:
+        list: A new list of integers sorted based on the criteria above.
+    """
+    sorted_ints = sorted(ints, key=lambda x: (bin(x).count('1'), x))
+    return sorted_ints
+
+
+def main():
+    # Convert input into integers
+    array = [int(n) for n in sys.argv[1:]]
+    result = sort_by_1_bits(array)
+    print(tuple(result))
+
+
+if __name__ == '__main__':
+    main()
diff --git a/challenge-271/sgreen/python/test.py b/challenge-271/sgreen/python/test.py
new file mode 100755
index 0000000000..0e2fcf2486
--- /dev/null
+++ b/challenge-271/sgreen/python/test.py
@@ -0,0 +1,26 @@
+#!/usr/bin/env python3
+
+import unittest
+ch_1 = __import__('ch-1')
+ch_2 = __import__('ch-2')
+
+
+class TestClass(unittest.TestCase):
+    def test_ch_1(self):
+        self.assertEqual(ch_1.maximum_ones([[0, 1],[1, 0]]), 1)
+        self.assertEqual(ch_1.maximum_ones([[0, 0, 0],[1, 0, 1]]), 2)
+        self.assertEqual(ch_1.maximum_ones([[0, 0],[1, 1],[0, 0]]), 2)
+
+    def test_ch_2(self):
+        self.assertEqual(
+            ch_2.sort_by_1_bits([0, 1, 2, 3, 4, 5, 6, 7, 8]),
+            [0, 1, 2, 4, 8, 3, 5, 6, 7]
+        )
+        self.assertEqual(
+            ch_2.sort_by_1_bits([1024, 512, 256, 128, 64]),
+            [64, 128, 256, 512, 1024]
+        )
+
+
+if __name__ == '__main__':
+    unittest.main()