动态规划

Leetcode 221 最大正方形

2018-08-06  本文已影响664人  LENN123

题目传送门

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

示例:

输入:

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

输出: 4

分析:

Python3代码:

class Solution:
    def maximalSquare(self, matrix):
        """
        :type matrix: List[List[str]]
        :rtype: int
        """
        if matrix == []: return 0 
        M, N = len(matrix), len(matrix[0])
        dp = [[0 for _ in range(N)]for _ in range(M)]
        Max = 0
        for i in range(N):   # dp矩阵初始化
            dp[0][i] = int(matrix[0][i])
            Max = max(dp[0][i], Max)
        for i in range(M):   # dp矩阵初始化
            dp[i][0] = int(matrix[i][0])
            Max = max(dp[i][0], Max)
        for i in range(1, M):
            for j in range(1, N):
                if matrix[i][j] == '0': continue
                dp[i][j] = min(dp[i-1][j],dp[i][j-1],dp[i-1][j-1])+1
                Max = max(dp[i][j], Max)
        return Max**2
上一篇 下一篇

猜你喜欢

热点阅读