在 C++ 中查找和为 k 的最大面积矩形子矩阵
假设我们有一个二维矩阵 mat 和一个值 K,我们需要找到和等于 K 的最长矩形子矩阵。
因此,如果输入类似于
2 | 8 | -5 | 6 |
-7 | 7 | 8 | -3 |
11 | -14 | 4 | 3 |
-4 | 3 | 1 | 10 |
并且 K = 9
则输出将是左上角点为 (1, 0) 且右下角点为 (3, 2)。
-7 | 7 | 8 |
11 | -14 | 4 |
-4 | 3 | 1 |
要解决此问题,我们将遵循以下步骤:
MAX := 100
定义一个函数 sum_k(),它将接收一个数组 arr、起始位置 start、结束位置 end、数组长度 n 和 k 作为参数。
定义一个 map
sum := 0,maximum_length := 0
对于 i 从 0 开始,当 i < n 时,更新 (i 增加 1),执行以下操作:
sum := sum + arr[i]
如果 sum 等于 k,则:
maximum_length := i + 1
start := 0
end := i
如果 sum 不在 map 中,则:
map[sum] := i
如果 (sum - k) 在 map 中,则:
如果 maximum_length < (i - map[sum - k]),则:
maximum_length := i - map[sum - k]
start := map[sum - k] + 1
end := i
当 maximum_length 不为 0 时返回 true
从主方法中执行以下操作:
row := mat 的行数,col := mat 的列数
定义一个大小为 row 的数组 temp。
定义一个数组 final_point = {0,0,0,0}
maxArea := -inf
对于 left 从 0 开始,当 left < col 时,更新 (left 增加 1),执行以下操作:
用 0 填充 temp
对于 right 从 left 开始,当 right < col 时,更新 (right 增加 1),执行以下操作:
对于 i 从 0 开始,当 i < row 时,更新 (i 增加 1),执行以下操作:
temp[i] := temp[i] + mat[i, right]
sum := sum_k(temp, up, down, row, k)
area := (down - up + 1) * (right - left + 1)
如果 sum 不为零且 maxArea < area,则:
final_point[0] := up,final_point[1] := down
final_point[2] := left,final_point[3] := right
maxArea := area
如果 final_point 为 [0,0,0,0] 且 mat[0,0] 不等于 k,则
返回“未找到子矩阵”
显示左上角点 (final_point[0], final_point[2])
显示右下角点 (final_point[1], final_point[3])
显示 mat 元素。
示例
让我们看看以下实现以获得更好的理解:
#include <bits/stdc++.h> using namespace std; const int MAX = 100; bool sum_k(int arr[], int& start, int& end, int n, int k) { unordered_map<int, int> map; int sum = 0, maximum_length = 0; for (int i = 0; i < n; i++) { sum += arr[i]; if (sum == k) { maximum_length = i + 1; start = 0; end = i; } if (map.find(sum) == map.end()) map[sum] = i; if (map.find(sum - k) != map.end()) { if (maximum_length < (i - map[sum - k])) { maximum_length = i - map[sum - k]; start = map[sum - k] + 1; end = i; } } } return (maximum_length != 0); } void sum_zero(vector<vector<int>> &mat, int k) { int row = mat.size(); int col = mat[0].size(); int temp[row], area; bool sum; int up, down; vector<int> final_point = {0,0,0,0}; int maxArea = INT_MIN; for (int left = 0; left < col; left++) { memset(temp, 0, sizeof(temp)); for (int right = left; right < col; right++) { for (int i = 0; i < row; i++) temp[i] += mat[i][right]; sum = sum_k(temp, up, down, row, k); area = (down - up + 1) * (right - left + 1); if (sum && maxArea < area) { final_point[0] = up; final_point[1] = down; final_point[2] = left; final_point[3] = right; maxArea = area; } } } if (final_point[0] == 0 && final_point[1] == 0 && final_point[2] == 0 && final_point[3] == 0 && mat[0][0] != k) { cout << "No sub-matrix found"; return; } cout << "(Top, Left) Coordinate: " << "(" << final_point[0] << ", " << final_point[2] << ")" << endl; cout << "(Bottom, Right) Coordinate: " << "(" << final_point[1] << ", " << final_point[3] << ")" << endl; for (int j = final_point[0]; j <= final_point[1]; j++) { for (int i = final_point[2]; i <= final_point[3]; i++) cout << mat[j][i] << " "; cout << endl; } } main(){ vector<vector<int>> v = { { 2, 8, -5, 6 }, { -7, 7, 8, -3 }, { 11, -14, 4, 3 }, { -4, 3, 1, 10 }}; sum_zero(v, 9); }
输入
{{ 2, 8, -5, 6 }, { -7, 7, 8, -3 }, { 11, -14, 4, 3 }, { -4, 3, 1, 10 }}, 9
输出
(Top, Left) Coordinate: (1, 0) (Bottom, Right) Coordinate: (3, 2) -7 7 8 11 -14 4 -4 3 1