leetcode 221 最大正方形 + 1277 统计全为1的正方形子矩阵

题目

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

示例

输入:matrix = [["1","0","1","0","0"],["1","0","1","1","1"],["1","1","1","1","1"],["1","0","0","1","0"]]

输出:4

解析

题外话,首先注意下函数签名:func maximalSquare(matrix [][]byte) int {}

这道题还是用动规五部曲来处理下

1.dp数组及其含义:

dp[i][j]:代码下标为i-1,j-1位置为右下角的正方形,最大面积为dp[i][j]。这个dp公式的定义很重要,首先是定义成了右下角,其次还用到了之前-1的这种方法,写代码会简单些

2.递推公式

if matrix[i-1][j-1] == '1' {

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

}

大致的思路是,首先要右下角的这个位置是1,否则就没啥用了,肯定不满足;在是1的前提下,类似木桶原理,右下角位置的最长边长,取决于另外三个位置的最小距离,然后+1

3.初始化

使用了-1的策略后,就是不需要特别的初始化了,默认是0

go 复制代码
func maximalSquare(matrix [][]byte) int {
   if len(matrix) == 0 || len(matrix[0]) == 0 {
       return 0
   }
   m := len(matrix)
   n := len(matrix[0])
   maxSide := 0
   dp := make([][]int, m+1)
   for i := 0; i <= m; i++ {
       dp[i] = make([]int, n+1)
   }

   for i := 1; i <= m; i++ {
       for j := 1; j <= n; j++ {
           if matrix[i-1][j-1] == '1' {
               dp[i][j] = min(dp[i-1][j-1], min(dp[i-1][j], dp[i][j-1])) + 1
               maxSide = max(maxSide, dp[i][j])
           }
       }
   }

   return maxSide * maxSide
}

func min(a, b int) int {
   if a > b {
       return b
   }
   return a
}

func max(a, b int) int {
   if a > b {
       return a
   }
   return b
}

1277 统计全为1的正方形子矩阵

题目

给你一个 m * n 的矩阵,矩阵中的元素不是 0 就是 1,请你统计并返回其中完全由 1 组成的 正方形 子矩阵的个数。

示例

输入:matrix =

\[0,1,1,1\], \[1,1,1,1\], \[0,1,1,1

]

输出:15

解释:

边长为 1 的正方形有 10 个。

边长为 2 的正方形有 4 个。

边长为 3 的正方形有 1 个。

正方形的总数 = 10 + 4 + 1 = 15.

解析

这道题和上面那道基本一样的思路,记住递推公式把

go 复制代码
func countSquares(matrix [][]int) int {
    if len(matrix) == 0 || len(matrix[0]) == 0 {
        return 0
    }
    m := len(matrix)
    n := len(matrix[0])
    dp := make([][]int, m+1)
    for i := 0; i <= m; i++ {
        dp[i] = make([]int, n+1)
    }
    res := 0
    for i := 1; i <= m; i++ {
        for j := 1; j <= n; j++ {
            if matrix[i-1][j-1] == 1 {
                dp[i][j] = min(dp[i-1][j-1], min(dp[i-1][j], dp[i][j-1])) + 1
                res += dp[i][j]
            }
        }
    }
    return res
}

func min(a, b int) int {
    if a > b {
        return b
    }
    return a
}
相关推荐
一百天成为python专家7 分钟前
K-近邻算法
数据结构·python·算法·pandas·近邻算法·ipython·python3.11
满分观察网友z25 分钟前
告别烦人的“三连发”:我的智能评论系统过滤之旅(1957. 删除字符使字符串变好)
算法
小新学习屋27 分钟前
《剑指offer》-数据结构篇-哈希表/数组/矩阵/字符串
数据结构·leetcode·哈希表
满分观察网友z28 分钟前
滑动窗口下的极限挑战:我在实时数据流中挖掘最大价值分(1695. 删除子数组的最大得分)
算法
PerfumerKarma28 分钟前
【WebGPU学习杂记】数学基础拾遗(2)变换矩阵中的齐次坐标推导与几何理解
学习·线性代数·矩阵
山烛1 小时前
KNN 算法中的各种距离:从原理到应用
人工智能·python·算法·机器学习·knn·k近邻算法·距离公式
guozhetao1 小时前
【ST表、倍增】P7167 [eJOI 2020] Fountain (Day1)
java·c++·python·算法·leetcode·深度优先·图论
吃着火锅x唱着歌1 小时前
LeetCode 611.有效三角形的个数
算法·leetcode·职场和发展
技术卷1 小时前
详解力扣高频SQL50题之619. 只出现一次的最大数字【简单】
sql·leetcode·oracle
CHANG_THE_WORLD4 小时前
金字塔降低采样
算法·金字塔采样