leetcode 221.最大正方形(python)

leetcode 221.最大正方形(python)

在一个由 0 和 1 组成的二维矩阵内，找到只包含 1 的最大正方形，并返回其面积。

示例:

输入: 

1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0

输出: 4

动态规划

dp方程:

dp(i, j) = min(dp(i-1, j) , dp(i, j-1), dp(i-1, j-1)) + 1

这个还是很容易想到的, 画个图自己推下就好

class Solution(object):
    def maximalSquare(self, matrix):
        # 动态规划
        if not matrix:
            return 0
        n1, n2 = len(matrix), len(matrix[0])
        dp = [[0 for _ in range(n2+1)] for _ in range(n1+1)]
        res = 0
        for i in range(1, n1+1):
            for j in range(1, n2+1):
                dp[i][j] = (min(dp[i-1][j], dp[i][j-1], dp[i-1][j-1]) + 1) * int(matrix[i-1][j-1])
                res = max(res, dp[i][j] ** 2)
        return res