Open In App

Print cells with same rectangular sums in a matrix

Improve
Improve
Like Article
Like
Save
Share
Report

Given a matrix of m x n matrix, we need to print all those cells at which sum of sub-matrix ended at this cell and sub-matrix starting at this cell is equal to remaining elements. For better understanding please see below diagram,

Examples : 

Input : mat[][] = {1, 2, 3, 5,
                   4, 1, 0, 2,
                   0, 1, 2, 0,
                   7, 1, 1, 0};
Output : (1, 1), (2, 2)      

In above matrix, cell (1, 1) and cell (2, 2)
are our required cells because,
For cell (1, 1), sum of red and green areas is same
1+2+4+1+0+2+1+2+0+1+1+0 = 3+5+0+7        
Same is true for cell (2, 2)
1+2+3+4+1+0+0+1+2+0+1+0 = 5+2+7+1    

We need to print all blue boundary cells for 
which sum of red area is equal to green area.

First we construct auxiliary sub-matrices similar to the linked article.

We construct two matrices sum[][] and sumr[][] such that sum[i][j] denotes sum of sub-matrix from mat[0][0] to mat[i][j]. And sumr for storing sum till last indices i.e. sumr[i][j] denotes sum of sub-matrix, mat[i][j] to mat[m – 1][n – 1]. 

Now we can use above matrices for solving this problem, red area shown in above diagram can be calculated by adding corresponding cells from sum and sumr matrices, as mat[i][j] is considered twice while calculating this sum we will subtract it once to get sum of red area. Getting sum of remaining element, the green part, is pretty easy, we will just subtract sum of red part from total sum of given matrix. 

So to check whether a particular cell satisfies the given condition we will calculate the sum of red part as explained above and compare it with total sum of matrix, if this sum is half of total sum, current cell satisfies the condition and hence a candidate for result.

Implementation:

C++




// C++ program to print cells with same rectangular
// sum diagonally
#include <bits/stdc++.h>
using namespace std;
#define R 4
#define C 4
 
// Method prints cell index at which rectangular sum is
// same at prime diagonal and other diagonal
void printCellWithSameRectangularArea(int mat[R][C],
                                     int m, int n)
{
    /* sum[i][j] denotes sum of sub-matrix, mat[0][0]
       to mat[i][j]
       sumr[i][j] denotes sum of sub-matrix, mat[i][j]
       to mat[m - 1][n - 1]  */
    int sum[m][n], sumr[m][n];
 
    //  Initialize both sum matrices by mat
    int totalSum = 0;
    for (int i = 0; i < m; i++)
    {
        for (int j = 0; j < n; j++)
        {
            sumr[i][j] = sum[i][j] = mat[i][j];
            totalSum += mat[i][j];
        }
    }
 
    //  updating first and last row separately
    for (int i = 1; i < m; i++)
    {
        sum[i][0] += sum[i-1][0];
        sumr[m-i-1][n-1] += sumr[m-i][n-1];
    }
 
    //  updating first and last column separately
    for (int j = 1; j < n; j++)
    {
        sum[0][j] += sum[0][j-1];
        sumr[m-1][n-j-1] += sumr[m-1][n-j];
    }
 
    //  updating sum and sumr indices by nearby indices
    for (int i = 1; i < m; i++)
    {
        for (int j = 1; j < n; j++)
        {
            sum[i][j] += sum[i-1][j] + sum[i][j-1] -
                                      sum[i-1][j-1];
            sumr[m-i-1][n-j-1] += sumr[m-i][n-j-1] +
                                  sumr[m-i-1][n-j] -
                                  sumr[m-i][n-j];
        }
    }
 
    //  Uncomment below code to print sum and reverse sum
    // matrix
    /*
        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                cout << sum[i][j] << " ";
            }
            cout << endl;
        }
        cout << endl;
        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                cout << sumr[i][j] << " ";
            }
            cout << endl;
        }
        cout << endl;    */
 
    /* print all those indices at which sum of prime diagonal
       rectangles is half of the total sum of matrix  */
    for (int i = 0; i < m; i++)
    {
        for (int j = 0; j < n; j++)
        {
            int mainDiagRectangleSum = sum[i][j] + sumr[i][j] -
                                                   mat[i][j];
            if (totalSum == 2 * mainDiagRectangleSum)
                cout << "(" << i << ", " << j << ")" << endl;
        }
    }
}
 
//  Driver code to test above methods
int main()
{
    int mat[R][C] =
    {
        1, 2, 3, 5,
        4, 1, 0, 2,
        0, 1, 2, 0,
        7, 1, 1, 0
    };
 
    printCellWithSameRectangularArea(mat, R, C);
 
    return 0;
}


Java




// Java program to print cells with
// same rectangular sum diagonally
 
class GFG {
     
static final int R = 4;
static final int C = 4;
 
// Method prints cell index at which
// rectangular sum is same at
// prime diagonal and other diagonal
static void printCellWithSameRectangularArea(int mat[][],
                                             int m, int n)
{
    /* sum[i][j] denotes sum of sub-matrix, mat[0][0]
    to mat[i][j]
    sumr[i][j] denotes sum of sub-matrix, mat[i][j]
    to mat[m - 1][n - 1] */
    int sum[][] = new int[m][n];
    int sumr[][] = new int[m][n];
 
    // Initialize both sum matrices by mat
    int totalSum = 0;
    for (int i = 0; i < m; i++) {
    for (int j = 0; j < n; j++) {
        sumr[i][j] = sum[i][j] = mat[i][j];
        totalSum += mat[i][j];
    }
    }
 
    // updating first and last row separately
    for (int i = 1; i < m; i++) {
    sum[i][0] += sum[i - 1][0];
    sumr[m - i - 1][n - 1] += sumr[m - i][n - 1];
    }
 
    // updating first and last column separately
    for (int j = 1; j < n; j++) {
    sum[0][j] += sum[0][j - 1];
    sumr[m - 1][n - j - 1] += sumr[m - 1][n - j];
    }
 
    // updating sum and sumr indices by nearby indices
    for (int i = 1; i < m; i++) {
    for (int j = 1; j < n; j++) {
        sum[i][j] += sum[i - 1][j] + sum[i][j - 1] -
                                     sum[i - 1][j - 1];
        sumr[m - i - 1][n - j - 1] += sumr[m - i][n - j - 1] +
                                      sumr[m - i - 1][n - j] -
                                      sumr[m - i][n - j];
    }
    }
 
    // Uncomment below code to print sum and reverse sum
    // matrix
    /*
        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                System.out.print( sum[i][j] + " ");
            }
            System.out.println();
        }
        System.out.println();
        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                System.out.print(sumr[i][j] + " ");
            }
            System.out.println();
        }
        System.out.println(); */
 
    /* print all those indices at which sum of prime diagonal
    rectangles is half of the total sum of matrix */
    for (int i = 0; i < m; i++) {
    for (int j = 0; j < n; j++) {
        int mainDiagRectangleSum = sum[i][j] +
                       sumr[i][j] - mat[i][j];
        if (totalSum == 2 * mainDiagRectangleSum)
        System.out.println("(" + i + ", " + j + ")");
    }
    }
}
 
// Driver code
public static void main(String[] args)
{
    int mat[][] = {{1, 2, 3, 5},
                   {4, 1, 0, 2},
                   {0, 1, 2, 0},
                   {7, 1, 1, 0}};
 
    printCellWithSameRectangularArea(mat, R, C);
}
}
 
// This code is contributed by Anant Agarwal.


Python3




# Python program to print cells with same rectangular
# sum diagonally
# R 4
# C 4
  
# Method prints cell index at which rectangular sum is
# same at prime diagonal and other diagonal
def printCellWithSameRectangularArea(mat, m, n):
    # sum[i][j] denotes sum of sub-matrix, mat[0][0]
    #  to mat[i][j]
    # sumr[i][j] denotes sum of sub-matrix, mat[i][j]
    # to mat[m - 1][n - 1] 
    sum = [[0 for i in range(m)]for j in range(n)]
    sumr = [[0 for i in range(m)]for j in range(n)]
  
    #  Initialize both sum matrices by mat
    totalSum = 0
    for i in range(m):
        for j in range(n):
            sumr[i][j] = sum[i][j] = mat[i][j];
            totalSum += mat[i][j]
  
    #  updating first and last row separately
    for i in range(1,m):
        sum[i][0] += sum[i-1][0]
        sumr[m-i-1][n-1] += sumr[m-i][n-1]
  
    #  updating first and last column separately
    for j in range(1,n):
        sum[0][j] += sum[0][j-1];
        sumr[m-1][n-j-1] += sumr[m-1][n-j]
  
    #  updating sum and sumr indices by nearby indices
    for i in range(1,m):
        for j in range(1,n):
            sum[i][j] += sum[i-1][j] + sum[i][j-1] - sum[i-1][j-1]
            sumr[m-i-1][n-j-1] += sumr[m-i][n-j-1] + sumr[m-i-1][n-j] - sumr[m-i][n-j]
  
    #  Uncomment below code to print sum and reverse sum
    # matrix
  
    # print all those indices at which sum of prime diagonal
    #   rectangles is half of the total sum of matrix 
    for i in range(m):
        for j in range(n):
            mainDiagRectangleSum = sum[i][j] + sumr[i][j] - mat[i][j]
            if (totalSum == 2 * mainDiagRectangleSum):
                print ("(",i,",",j,")")
  
#  Driver code to test above methods
mat =[[1, 2, 3, 5,],
     [4, 1, 0, 2,],
     [0, 1, 2, 0],
     [7, 1, 1, 0]]
printCellWithSameRectangularArea(mat, 4, 4)
 
# Contributed by Afzal


C#




// C# program to print cells with
// same rectangular sum diagonally
using System;
 
class GFG {
     
static int R = 4;
static int C = 4;
 
// Method prints cell index at which
// rectangular sum is same at
// prime diagonal and other diagonal
static void printCellWithSameRectangularArea(int [,]mat,
                                             int m, int n)
{
    /* sum[i][j] denotes sum of sub-
       matrix, mat[0][0] to mat[i][j]
       sumr[i][j] denotes sum of sub-matrix,
       mat[i][j] to mat[m - 1][n - 1] */
    int [,]sum = new int[m, n];
    int [,]sumr = new int[m, n];
 
    // Initialize both sum matrices by mat
    int totalSum = 0;
    for (int i = 0; i < m; i++) {
    for (int j = 0; j < n; j++) {
        sumr[i, j] = sum[i, j] = mat[i, j];
        totalSum += mat[i, j];
    }
    }
 
    // updating first and last row separately
    for (int i = 1; i < m; i++)
    {
    sum[i, 0] += sum[i - 1, 0];
    sumr[m - i - 1, n - 1] += sumr[m - i, n - 1];
    }
 
    // updating first and last column separately
    for (int j = 1; j < n; j++)
    {
    sum[0,j] += sum[0,j - 1];
    sumr[m - 1,n - j - 1] += sumr[m - 1,n - j];
    }
 
    // updating sum and sumr indices by nearby indices
    for (int i = 1; i < m; i++) {
    for (int j = 1; j < n; j++) {
        sum[i,j] += sum[i - 1,j] + sum[i,j - 1] -
                                   sum[i - 1,j - 1];
        sumr[m - i - 1,n - j - 1] += sumr[m - i,n - j - 1] +
                                     sumr[m - i - 1,n - j] -
                                     sumr[m - i,n - j];
    }
    }
 
    // Uncomment below code to print sum and reverse sum
    // matrix
    /*
        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                System.out.print( sum[i][j] + " ");
            }
            System.out.println();
        }
        System.out.println();
        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                System.out.print(sumr[i][j] + " ");
            }
            System.out.println();
        }
        System.out.println(); */
 
    /* print all those indices at which sum
       of prime diagonal rectangles is half
       of the total sum of matrix */
    for (int i = 0; i < m; i++) {
    for (int j = 0; j < n; j++) {
        int mainDiagRectangleSum = sum[i,j] +
                    sumr[i,j] - mat[i,j];
                     
        if (totalSum == 2 * mainDiagRectangleSum)
        Console.WriteLine("(" + i + ", " + j + ")");
    }
    }
}
 
// Driver code
public static void Main()
{
    int [,]mat = {{1, 2, 3, 5},
                  {4, 1, 0, 2},
                  {0, 1, 2, 0},
                  {7, 1, 1, 0}};
 
    printCellWithSameRectangularArea(mat, R, C);
}
}
 
// This code is contributed by vt_m.


Javascript




<script>
    // Javascript program to print cells with
    // same rectangular sum diagonally
     
    let R = 4;
    let C = 4;
 
    // Method prints cell index at which
    // rectangular sum is same at
    // prime diagonal and other diagonal
    function printCellWithSameRectangularArea(mat, m, n)
    {
        /* sum[i][j] denotes sum of sub-matrix, mat[0][0]
        to mat[i][j]
        sumr[i][j] denotes sum of sub-matrix, mat[i][j]
        to mat[m - 1][n - 1] */
        let sum = new Array(m);
        let sumr = new Array(m);
 
        // Initialize both sum matrices by mat
        let totalSum = 0;
        for (let i = 0; i < m; i++)
        {
          sum[i] = new Array(n);
          sumr[i] = new Array(n);
          for (let j = 0; j < n; j++)
          {
              sumr[i][j] = sum[i][j] = mat[i][j];
              totalSum += mat[i][j];
          }
        }
 
        // updating first and last row separately
        for (let i = 1; i < m; i++)
        {
          sum[i][0] += sum[i - 1][0];
          sumr[m - i - 1][n - 1] += sumr[m - i][n - 1];
        }
 
        // updating first and last column separately
        for (let j = 1; j < n; j++) {
          sum[0][j] += sum[0][j - 1];
          sumr[m - 1][n - j - 1] += sumr[m - 1][n - j];
        }
 
        // updating sum and sumr indices by nearby indices
        for (let i = 1; i < m; i++)
        {
          for (let j = 1; j < n; j++)
          {
              sum[i][j] += sum[i - 1][j] + sum[i][j - 1] -
                                           sum[i - 1][j - 1];
              sumr[m - i - 1][n - j - 1] += sumr[m - i][n - j - 1] +
                                            sumr[m - i - 1][n - j] -
                                            sumr[m - i][n - j];
          }
        }
 
        /* print all those indices at which sum of prime diagonal
        rectangles is half of the total sum of matrix */
        for (let i = 0; i < m; i++) {
          for (let j = 0; j < n; j++)
          {
              let mainDiagRectangleSum = sum[i][j] +
                             sumr[i][j] - mat[i][j];
              if (totalSum == 2 * mainDiagRectangleSum)
              document.write("(" + i + ", " + j + ")" + "</br>");
          }
        }
    }
     
    let mat = [[1, 2, 3, 5],
                [4, 1, 0, 2],
                [0, 1, 2, 0],
                [7, 1, 1, 0]];
   
    printCellWithSameRectangularArea(mat, R, C);
 
</script>


Output

(1, 1)
(2, 2)

Time complexity: O(n x m).
Auxiliary Space: O(n x m).

 



Last Updated : 06 Jul, 2022
Like Article
Save Article
Previous
Next
Share your thoughts in the comments
Similar Reads