【python】:最大黑方阵

给定一个方阵,其中每个单元(像素)非黑即白。设计一个算法,找出 4 条边皆为黑色像素的最大子方阵。

返回一个数组 [r, c, size] ,其中 r, c 分别代表子方阵左上角的行号和列号,size 是子方阵的边长。若有多个满足条件的子方阵,返回 r 最小的,若 r 相同,返回 c 最小的子方阵。若无满足条件的子方阵,返回空数组。

示例 1:

输入:
[
   [1,0,1],
   [0,0,1],
   [0,0,1]
]
输出: [1,0,2]
解释: 输入中 0 代表黑色,1 代表白色,标粗的元素即为满足条件的最大子方阵
示例 2:

输入:
[
   [0,1,1],
   [1,0,1],
   [1,1,0]
]
输出: [0,0,1]
提示:

matrix.length == matrix[0].length <= 200

一、动态规划
从右下角向左上角遍历,存储每个点右边和下面连续的0数;再从左上角向右下角遍历,选行和列最小的值,递减判断能组成的最大正方形。

class Solution:
    def findSquare(self, matrix: List[List[int]]) -> List[int]:
        if not matrix:
            return []
        mrow = collections.defaultdict(int)
        mcol = collections.defaultdict(int)
        len_row = len(matrix)
        len_col = len(matrix[0])
        res = []
        for r in range(len_row)[::-1]:
            for c in range(len_col)[::-1]:
                if matrix[r][c] == 0:
                    mrow[r, c] = 1 + mrow[r, c+1]
                    mcol[r, c] = 1 + mcol[r+1, c]
        for r in range(len_row):
            for c in range(len_col):
                if matrix[r][c] == 0:
                    max_size = min(mrow[r,c], mcol[r,c])
                    cur_size = 0 if not res else res[2]
                    for size in range(max_size, cur_size, -1):
                        if mcol[r, c+size-1] >= size and mrow[r+size-1,c] >= size:
                            res = [r, c, size]
                            break
        return res
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值