假设我们有一个二元矩阵。我们可以先根据需要多次重新排列列,然后找到返回仅包含 1 的最大子矩阵的面积。
所以,如果输入是这样的
1 | 0 | 0 |
1 | 1 | 1 |
1 | 0 | 1 |
那么输出将是 4,因为我们可以安排就像 -
1 | 0 | 0 |
1 | 1 | 1 |
1 | 1 | 0 |
让我们看看以下实现以获得更好的理解 -
def solve(matrix): n, m = len(matrix), len(matrix[0]) ans = 0 for i in range(1, n) : for j in range(m) : if matrix[i][j] : matrix[i][j] += matrix[i-1][j] for row in matrix : row.sort() for j in range(m-1, -1, -1): ans = max(ans, row[j] *(m - j)) return ans matrix = [ [1, 0, 0], [1, 1, 1], [1, 0, 1] ] print(solve(matrix))
[ [1, 0, 0], [1, 1, 1], [1, 0, 1] ]输出结果
4