[leetcode] 221. Maximal Square @ python

关注

原创最新推荐文章于 2026-11-01 12:14:11 发布 386 阅读 AI 写同款· GEO 优化 ›

电赛MSP430F5529之超声波测距+OLED 显示（含主要代码）目录前言一、超声波介绍二、部分代码三、总结题外话：前言哎，又双叒叕，电赛延期了，看起来像是遥遥无期，但是肯定要考虑大局，大家就继续抓紧时间学习知识，巩固自己的知识，更进一步。好了，今天主要是继承上一篇博文（OLED显示）写的，主要是写超声波测距的（US-015）一、超声波介绍利用TI公司的MSP430F5529单片机+超声波模块+OLED 实现这里主要是利用单片机定时器的捕获功能来捕获超声波高电平持续时间，再计算得到距离，最后再在OLED 屏幕上显示出来，上图阅读详情

原题

Given a 2D binary matrix filled with 0’s and 1’s, find the largest square containing only 1’s and return its area.

Example:

Input:

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

Output: 4

解法

参考: 花花酱 LeetCode 221. Maximal Square
动态规划. base case是当matrix长度为0时, 返回0. 构造dp数组, 将matrix里的字符串转化为数字, dp[i][j]表示以matrix[i][j]为右下角能构成的最大正方形的边长, 我们从第1行第1列开始, 求dp[i][j].

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

然后找出二维数组dp中最大的值就是最大正方形的边, 取平方即可.
Time: O(mn)
Space: O(mn)

代码

class Solution(object):
    def maximalSquare(self, matrix):
        """
        :type matrix: List[List[str]]
        :rtype: int
        """
        # base case
        if not matrix: return 0
        res = 0
        row, col = len(matrix), len(matrix[0])
        dp = [[1 if matrix[i][j] == '1' else 0 for j in range(col)] for i in range(row)]
        for i in range(1, row):
            for j in range(1, col):
                if matrix[i][j] == '1':
                    dp[i][j] = min(dp[i-1][j-1], dp[i][j-1], dp[i-1][j]) + 1
                else:
                    dp[i][j] = 0
                    
        for row in dp:
            res = max(res, max(row))
        return res*res