Create sort-the-matrix-diagonally.py

Author: kamyu104

Python/sort-the-matrix-diagonally.py

@@ -0,0 +1,22 @@
+# Time:  O(m * n * log(min(m, n))
+# Space: O(m * n)
+
+import collections
+
+
+class Solution(object):
+    def diagonalSort(self, mat):
+        """
+        :type mat: List[List[int]]
+        :rtype: List[List[int]]
+        """
+        lookup = collections.defaultdict(list)
+        for i in xrange(len(mat)):
+            for j in xrange(len(mat[0])):
+                lookup[i-j].append(mat[i][j])
+        for v in lookup.itervalues():
+            v.sort()
+        for i in reversed(xrange(len(mat))):
+            for j in reversed(xrange(len(mat[0]))):
+                mat[i][j] = lookup[i-j].pop()
+        return mat