Create delete-columns-to-make-sorted-ii.py

Author: kamyu104

Python/delete-columns-to-make-sorted-ii.py

@@ -0,0 +1,40 @@
+# Time:  O(n * m)
+# Space: O(n)
+
+class Solution(object):
+    def minDeletionSize(self, A):
+        """
+        :type A: List[str]
+        :rtype: int
+        """
+        result = 0
+        unsorted = set(range(len(A)-1))
+        for j in xrange(len(A[0])):
+            if any(A[i][j] > A[i+1][j] for i in unsorted):
+                result += 1
+            else:
+                unsorted -= set(i for i in unsorted if A[i][j] < A[i+1][j])
+        return result
+
+
+# Time:  O(n * m)
+# Space: O(n)
+class Solution2(object):
+    def minDeletionSize(self, A):
+        """
+        :type A: List[str]
+        :rtype: int
+        """
+        result = 0
+        is_sorted = [False]*(len(A)-1)
+        for j in xrange(len(A[0])):
+            tmp = is_sorted[:]
+            for i in xrange(len(A)-1):
+                if A[i][j] > A[i+1][j] and tmp[i] == False:
+                    result += 1
+                    break
+                if A[i][j] < A[i+1][j]:
+                    tmp[i] = True
+            else:
+                is_sorted = tmp
+        return result