5454. 統計全 1 子矩形
給你一個只包含 0 和 1 的 rows * columns
矩陣 mat
,請你返回有多少個 子矩形 的元素全部都是 1 。
示例 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 。
示例 2:
輸入:mat = [[0,1,1,0], [0,1,1,1], [1,1,1,0]] 輸出:24 解釋: 有 8 個 1x1 的子矩形。 有 5 個 1x2 的子矩形。 有 2 個 1x3 的子矩形。 有 4 個 2x1 的子矩形。 有 2 個 2x2 的子矩形。 有 2 個 3x1 的子矩形。 有 1 個 3x2 的子矩形。 矩形數目總共 = 8 + 5 + 2 + 4 + 2 + 2 + 1 = 24 。
示例 3:
輸入:mat = [[1,1,1,1,1,1]] 輸出:21
示例 4:
輸入:mat = [[1,0,1],[0,1,0],[1,0,1]] 輸出:5
提示:
1 <= rows <= 150
1 <= columns <= 150
0 <= mat[i][j] <= 1
class Solution {
int mymin(int a, int b) {
return a < b ? a : b;
}
public:
int up[151][151] = { 0 };
int left[151][151] = { 0 };
int numSubmat(vector<vector<int>>& mat) {
int result = 0;
for (int i = 0; i < mat.size(); ++i) {
for (int j = 0; j < mat[0].size(); ++j) {
if (mat[i][j]) {
up[i + 1][j + 1] = up[i + 1][j] + 1;
left[i + 1][j + 1] = left[i][j + 1] + 1;
int l = left[i + 1][j + 1];
int u = up[i + 1][j + 1];
if (l < u) {
int re = u - 1;
for (int k = i; k > i + 1 - l; --k) {
if (up[k][j] < re) {
result += mymin(up[k][j], u - 1);
re = mymin(up[k][j], u - 1);
}
else {
result += re;
}
}
}
else {
int re = l - 1;
for (int k = j; k > j + 1 - u; --k) {
if (left[i][k] < re) {
result += mymin(left[i][k], l - 1);
re = mymin(left[i][k], l - 1);
}
else {
result += re;
}
}
}
result += l + u - 1;
}
}
}
return result;
}
};