前言
最近被我大哥安利了一道算法题, 这道题说难, 还不至于我做不出来, 说简单吧, 我还想不到最优解, 等把最优解告诉我之后, 我还正好能理解. 我甚至曾经怯怯的认为, 这题就是我哥专门给我找的, 嘿嘿, 心中说不出的小欢喜.
题来了, 此题出自力扣, 原题链接:
https://leetcode-cn.com/problems/count-submatrices-with-all-ones/
描述: 给你一个只包含 0 和 1 的 rows * columns
矩阵 mat
,请你返回有多少个 子矩形 的元素全部都是 1 。
例子:
输入:mat = [[1,0,1],
[1,1,0],
[1,1,0]]
输出:13
解释:
有 6 个 1x1 的矩形。
有 2 个 1x2 的矩形。
有 3 个 2x1 的矩形。
有 1 个 2x2 的矩形。
有 1 个 3x1 的矩形。
矩形数目总共 = 6 + 2 + 3 + 1 + 1 = 13 。
题意清晰明了, 开始尝试解题(使用 C 来进行解题).
方案一
首先直观上最先想到的, 就是穷举了. 一力破十会. 将所有出现的情况遍历一遍, 然后就能得出总数了. 思路如下:
- 利用
i
,j
将二维数组的所有节点遍历一遍 - 利用
m
,n
将以[i][j]
为左上顶点的子矩阵遍历一遍 - 判断
i
,j