在 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
数据结构
网络
关系型数据库管理系统
操作系统
Java
iOS
HTML
CSS
Android
Python
C 编程
C++
C#
MongoDB
MySQL
Javascript
PHP