在 C++ 中查找和为 k 的最大面积矩形子矩阵


假设我们有一个二维矩阵 mat 和一个值 K,我们需要找到和等于 K 的最长矩形子矩阵。

因此,如果输入类似于

28-56
-778-3
11-1443
-43110

并且 K = 9

则输出将是左上角点为 (1, 0) 且右下角点为 (3, 2)。

-778
11-144
-431

要解决此问题,我们将遵循以下步骤:

  • 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

更新于:2020 年 8 月 20 日

206 次浏览

开启你的 职业生涯

通过完成课程获得认证

开始学习
广告