📜  总和为0的最大矩形子矩阵

📅  最后修改于: 2021-05-06 18:44:12             🧑  作者: Mango

给定2D矩阵,找到总和为0的最大矩形子矩阵。例如,考虑以下N x M输入矩阵

例子:

Input :  1,  2,  3
        -3, -2, -1          
         1,  7,  5

Output : 1,  2,  3          
        -3, -2, -1

Input :  9,  7, 16,  5
         1, -6, -7,  3          
         1,  8,  7,  9          
         7, -2,  0, 10

Output :-6, -7
          8,  7          
         -2,  0    

这个问题的幼稚解决方案是检查给定2D数组中的每个可能的矩形。此解决方案需要4个嵌套循环,并且此解决方案的时间复杂度为O(n ^ 4)。

该解决方案基于2D矩阵中的最大和矩形。这个想法是将问题减少到一维数组。我们可以使用哈希在O(n)时间内找到一维数组中子数组的最大长度。我们一一固定左列和右列,并为每个左列和右列对找到最大有0个连续行的子数组。我们基本上为每个固定的左右列对找到顶部和底部的行号(总和为零)。要找到顶部和底部的行号,请从左到右计算每行中的元素总和,并将这些总和存储在一个名为temp []的数组中。因此,temp [i]表示第i行中从左到右的元素总和。如果我们找到最大的子数组,其temp的总和为0,否。的元素数大于以前的数字。元素,然后更新final row_up,final row_down,final col_left,final col_right的值。

// A C++ program to find Largest rectangular
// sub-matrix whose sum is 0
#include 
using namespace std;
  
const int MAX = 100;
  
// This function basically finds largest 0
// sum aubarray in temp[0..n-1]. If 0 sum
// does't exist, then it returns false. Else
// it returns true and sets starting and
// ending indexes as starti and endj.
bool sumZero(int temp[], int* starti,
            int* endj, int n)
{
    // Map to store the previous sums
    map presum;
    int sum = 0; // Initialize sum of elements
  
    // Initialize length of sub-array with sum 0
    int max_length = 0;
  
    // Traverse through the given array
    for (int i = 0; i < n; i++)
    {
        // Add current element to sum
        sum += temp[i];
  
        if (temp[i] == 0 && max_length == 0)
        {
            *starti = i;
            *endj = i;
            max_length = 1;
        }
        if (sum == 0)
        {
            if (max_length < i + 1)
            {
                *starti = 0;
                *endj = i;
            }
            max_length = i + 1;
        }
  
        // Look for this sum in Hash table
        if (presum.find(sum) != presum.end())
        {
            // store previous max_length so
            // that we can check max_length
            // is updated or not
            int old = max_length;
  
            // If this sum is seen before,
            // then update max_len
            max_length = max(max_length, i - presum[sum]);
  
            if (old < max_length)
            {
                // If max_length is updated then
                // enter and update start and end
                // point of array
                *endj = i;
                *starti = presum[sum] + 1;
            }
        }
        else
  
            // Else insert this sum with
            // index in hash table
            presum[sum] = i;
    }
  
    // Return true if max_length is non-zero
    return (max_length != 0);
}
  
// The main function that finds Largest rectangle
// sub-matrix in a[][] whose sum is 0.
void sumZeroMatrix(int a[][MAX], int row, int col)
{
    int temp[row];
  
    // Variables to store the final output
    int fup = 0, fdown = 0, fleft = 0, fright = 0;
    int sum;
    int up, down;
    int maxl = INT_MIN;
  
    // Set the left column
    for (int left = 0; left < col; left++)
    {
        // Initialize all elements of temp as 0
        memset(temp, 0, sizeof(temp));
  
        // Set the right column for the left column
        // set by outer loop
        for (int right = left; right < col; right++)
        {
            // Calculate sum between current left
            // and right for every row 'i'
            for (int i = 0; i < row; i++)
                temp[i] += a[i][right];
  
            // Find largest subarray with 0 sum in
            // temp[]. The sumZero() function also
            // sets values of start and finish. So
            // 'sum' is sum of rectangle between (start,
            // left) and (finish, right) which is
            // boundary columns strictly as left and right.
            bool sum = sumZero(temp, &up, &down, row);
            int ele = (down - up + 1) * (right - left + 1);
  
            // Compare no. of elements with previous
            // no. of elements in sub-Matrix.
            // If new sub-matrix has more elements
            // then update maxl and final boundaries
            // like fup, fdown, fleft, fright
            if (sum && ele > maxl)
            {
                fup = up;
                fdown = down;
                fleft = left;
                fright = right;
                maxl = ele;
            }
        }
    }
  
    // If there is no change in boundaries
    // than check if a[0][0] is 0
    // If it not zero then print 
    // that no such zero-sum sub-matrix exists
    if (fup == 0 && fdown == 0 && fleft == 0 &&
            fright == 0 && a[0][0] != 0) {
        cout << "No zero-sum sub-matrix exists";
        return;
    }
  
    // Print final values
    for (int j = fup; j <= fdown; j++)
    {
        for (int i = fleft; i <= fright; i++)
            cout << a[j][i] << " ";
        cout << endl;
    }
}
  
// Driver program to test above functions
int main()
{
    int a[][MAX] = { { 9, 7, 16, 5 }, { 1, -6, -7, 3 },
                      { 1, 8, 7, 9 }, { 7, -2, 0, 10 } };
   
    int row = 4, col = 4;
    sumZeroMatrix(a, row, col);
    return 0;
}

输出:

-6, -7
     8, 7          
    -2, 0