Merge pull request #7863 from spadacciniweb/PWC-211

PWC 211

1 files changed, 43 insertions, 0 deletions

diff --git a/challenge-211/spadacciniweb/python/ch-1.py b/challenge-211/spadacciniweb/python/ch-1.py
new file mode 100644
index 0000000000..00f663b9f1
--- /dev/null
+++ b/challenge-211/spadacciniweb/python/ch-1.py
@@ -0,0 +1,43 @@
+# Task 1: Toeplitz Matrix
+# Submitted by: Mohammad S Anwar
+# 
+# You are given a matrix m x n.
+# Write a script to find out if the given matrix is Toeplitz Matrix.
+# A matrix is Toeplitz if every diagonal from top-left to bottom-right has the same elements.
+# 
+# 
+# Example 1
+# Input: @matrix = [ [4, 3, 2, 1],
+#                    [5, 4, 3, 2],
+#                    [6, 5, 4, 3],
+#                  ]
+# Output: true
+# 
+# Example 2
+# Input: @matrix = [ [1, 2, 3],
+#                    [3, 2, 1],
+#                  ]
+# Output: false
+
+import numpy
+
+def check_toeplitz(matrix):
+    m = numpy.array(matrix)
+    dim = min(m.shape)
+    value = m[0][0]
+    for i in range(dim):
+        if m[i][i] != value:
+            return 'false'
+    return 'true'
+
+if __name__ == "__main__":
+    matrix = [ [4, 3, 2, 1],
+               [5, 4, 3, 2],
+               [6, 5, 4, 3],
+             ]
+    print(check_toeplitz(matrix))
+
+    matrix = [ [1, 2, 3],
+               [3, 2, 1],
+             ]
+    print(check_toeplitz(matrix))