Create matrix-similarity-after-cyclic-shifts.py

Author: kamyu104

Python/matrix-similarity-after-cyclic-shifts.py

@@ -0,0 +1,12 @@
+# Time:  O(m * n)
+# Space: O(1)
+
+# array
+class Solution(object):
+    def areSimilar(self, mat, k):
+        """
+        :type mat: List[List[int]]
+        :type k: int
+        :rtype: bool
+        """
+        return all(row[i] == row[(i+k)%len(row)]for row in mat for i in xrange(len(row)))