2021-03-20:给定一个二维数组matrix,其中的值不是0就是1,返回全部由1组成的子矩形数量。
2021-03-20:给定一个二维数组matrix,其中的值不是0就是1,返回全部由1组成的子矩形数量。
福大大 答案2021-03-20:
按行遍历二维数组,构造直方图。
单调栈,大压小。有代码。
代码用golang编写,代码如下:
package main
import "fmt"
func main() {
matrix := [][]int{
{1, 1, 1, 1, 1, 1},
}
ret := numSubmat(matrix)
fmt.Println(ret)
}
func numSubmat(mat [][]int) int {
if len(mat) == 0 || len(mat[0]) == 0 {
return 0
}
nums := 0
height := make([]int, len(mat[0]))
for i := 0; i < len(mat); i++ {
for j := 0; j < len(mat[0]); j++ {
if mat[i][j] == 0 {
height[j] = 0
} else {
height[j]++
}
}
nums += countFromBottom(height)
}
return nums
}
func countFromBottom(height []int) int {
if len(height) == 0 {
return 0
}
nums := 0
stack := make([]int, len(height))
si := -1
for i := 0; i < len(height); i++ {
for si != -1 && height[stack[si]] >= height[i] {
cur := stack[si]
si--
if height[cur] > height[i] {
left := -1
if si != -1 {
left = stack[si]
}
n := i - left - 1
heightleft := 0
if left != -1 {
heightleft = height[left]
}
down := getMax(heightleft, height[i])
nums += (height[cur] - down) * num(n)
}
}
si++
stack[si] = i
}
for si != -1 {
cur := stack[si]
si--
left := -1
if si != -1 {
left = stack[si]
}
n := len(height) - left - 1
down := 0
if left != -1 {
down = height[left]
}
nums += (height[cur] - down) * num(n)
}
return nums
}
func num(n int) int {
return (n * (1 + n)) >> 1
}
func getMax(a int, b int) int {
if a > b {
return a
} else {
return b
}
}
执行结果如下:
公众号:福大大架构师每日一题
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
· AI与.NET技术实操系列(二):开始使用ML.NET
· 记一次.NET内存居高不下排查解决与启示
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· AI技术革命,工作效率10个最佳AI工具