diff options
Diffstat (limited to 'challenge-266/sgreen/python/ch-2.py')
| -rwxr-xr-x | challenge-266/sgreen/python/ch-2.py | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/challenge-266/sgreen/python/ch-2.py b/challenge-266/sgreen/python/ch-2.py new file mode 100755 index 0000000000..d0df7257ee --- /dev/null +++ b/challenge-266/sgreen/python/ch-2.py @@ -0,0 +1,45 @@ +#!/usr/bin/env python3 + +import json +import sys + + +def x_matrix(matrix: list) -> bool: + """Determine if the matrix is a X Matrix + + Args: + matrix (list): The supplied matrix + + Returns: + bool: Wether the matrix as a X Matrix or not + """ + rows = len(matrix) + + # Check we have a square + for row in range(rows): + if len(matrix[row]) != rows: + raise ValueError("Please specify a square matrix") + + # Check that all the values are correct + for row in range(rows): + for col in range(rows): + if col == row or col == rows - 1 - row: + # We are expecting a non-zero value + if matrix[row][col] == 0: + return False + elif matrix[row][col] != 0: + # We are expecting a zero value + return False + + return True + + +def main(): + # Parse the matrix from the input + matrix = json.loads(sys.argv[1]) + result = x_matrix(matrix) + print('true' if result else 'false') + + +if __name__ == '__main__': + main() |
