Create remove-all-ones-with-row-and-column-flips.py

Author: kamyu104

Python/remove-all-ones-with-row-and-column-flips.py

@@ -0,0 +1,10 @@
+# Time:  O(m * n)
+# Space: O(1)
+
+class Solution(object):
+    def removeOnes(self, grid):
+        """
+        :type grid: List[List[int]]
+        :rtype: bool
+        """
+        return all(grid[i] == grid[0] or all(grid[i][j] != grid[0][j] for j in xrange(len(grid[0]))) for i in xrange(1, len(grid)))