Given an n x n square matrix, find sum of all sub-squares of size k x k

Given an n x n square matrix, find sum of all sub-squares of size k x k where k is smaller than or equal to n.

Examples :

Input:
n = 5, k = 3
arr[][] = { {1, 1, 1, 1, 1},
            {2, 2, 2, 2, 2},
            {3, 3, 3, 3, 3},
            {4, 4, 4, 4, 4},
            {5, 5, 5, 5, 5},
         };
Output:
       18  18  18
       27  27  27
       36  36  36


Input:
n = 3, k = 2
arr[][] = { {1, 2, 3},
            {4, 5, 6},
            {7, 8, 9},
         };
Output:
       12  16
       24  28

A Simple Solution is to one by one pick starting point (leftmost-topmost corner) of all possible sub-squares. Once the starting point is picked, calculate sum of sub-square starting with the picked starting point.

Following is the implementation of this idea.

C++

filter_none

edit
close

play_arrow

link
brightness_4
code

// A simple C++ program to find sum of all subsquares of size k x k
#include <iostream>
using namespace std;
  
// Size of given matrix
#define n 5
  
// A simple function to find sum of all sub-squares of size k x k
// in a given square matrix of size n x n
void printSumSimple(int mat[][n], int k)
{
   // k must be smaller than or equal to n
   if (k > n) return;
  
   // row number of first cell in current sub-square of size k x k
   for (int i=0; i<n-k+1; i++)
   {
      // column of first cell in current sub-square of size k x k
      for (int j=0; j<n-k+1; j++)
      {
          // Calculate and print sum of current sub-square
          int sum = 0;
          for (int p=i; p<k+i; p++)
             for (int q=j; q<k+j; q++)
                 sum += mat[p][q];
           cout << sum << "  ";
      }
  
      // Line separator for sub-squares starting with next row
      cout << endl;
   }
}
  
// Driver program to test above function
int main()
{
    int mat[n][n] = {{1, 1, 1, 1, 1},
                     {2, 2, 2, 2, 2},
                     {3, 3, 3, 3, 3},
                     {4, 4, 4, 4, 4},
                     {5, 5, 5, 5, 5},
                    };
    int k = 3;
    printSumSimple(mat, k);
    return 0;
}

chevron_right


Java

filter_none

edit
close

play_arrow

link
brightness_4
code

// A simple Java program to find sum of all 
// subsquares of size k x k
class GFG
{
      
    // Size of given matrix
    static final int n = 5;
      
    // A simple function to find sum of all 
    //sub-squares of size k x k in a given 
    // square matrix of size n x n
    static void printSumSimple(int mat[][], int k)
    {
  
        // k must be smaller than or 
        // equal to n
        if (k > n) return;
          
        // row number of first cell in 
        // current sub-square of size k x k
        for (int i = 0; i < n-k+1; i++)
        {
              
            // column of first cell in current 
            // sub-square of size k x k
            for (int j = 0; j < n-k+1; j++)
            {
                  
                // Calculate and print sum of 
                // current sub-square
                int sum = 0;
                for (int p = i; p < k+i; p++)
                    for (int q = j; q < k+j; q++)
                        sum += mat[p][q];
  
                System.out.print(sum+ " ");
            }
          
            // Line separator for sub-squares 
            // starting with next row
            System.out.println();
        }
    }
      
    // Driver Program to test above function
    public static void main(String arg[])
    {
        int mat[][] = {{1, 1, 1, 1, 1},
                       {2, 2, 2, 2, 2},
                       {3, 3, 3, 3, 3},
                       {4, 4, 4, 4, 4},
                       {5, 5, 5, 5, 5}};
        int k = 3;
        printSumSimple(mat, k);
    }
}
  
// This code is contributed by Anant Agarwal.

chevron_right


C#

filter_none

edit
close

play_arrow

link
brightness_4
code

// A simple C# program to find sum of all 
// subsquares of size k x k
using System;
  
class GFG
{
    // Size of given matrix
    static int n = 5;
      
    // A simple function to find sum of all 
    //sub-squares of size k x k in a given 
    // square matrix of size n x n
    static void printSumSimple(int [,]mat, int k)
    {
        // k must be smaller than or 
        // equal to n
        if (k > n) return;
          
        // row number of first cell in 
        // current sub-square of size k x k
        for (int i = 0; i < n-k+1; i++)
        {
            // column of first cell in current 
            // sub-square of size k x k
            for (int j = 0; j < n-k+1; j++)
            {
                // Calculate and print sum of 
                // current sub-square
                int sum = 0;
                for (int p = i; p < k+i; p++)
                    for (int q = j; q < k+j; q++)
                        sum += mat[p,q];
  
                Console.Write(sum+ " ");
            }
          
            // Line separator for sub-squares 
            // starting with next row
            Console.WriteLine();
        }
    }
      
    // Driver Program to test above function
    public static void Main()
    {
        int [,]mat = {{1, 1, 1, 1, 1},
                      {2, 2, 2, 2, 2},
                      {3, 3, 3, 3, 3},
                      {4, 4, 4, 4, 4},
                      {5, 5, 5, 5, 5}};
        int k = 3;
        printSumSimple(mat, k);
    }
}
  
// This code is contributed by Sam007

chevron_right


PHP

filter_none

edit
close

play_arrow

link
brightness_4
code

<?php
// A simple PHP program to find 
// sum of all subsquares of size
// k x k
  
// Size of given matrix
$n = 5;
  
// function to find sum of all sub - 
// squares of size k x k in a given 
// square matrix of size n x n
function printSumSimple( $mat, $k)
{
    global $n;
      
    // k must be smaller than
    // or equal to n
    if ($k > $n) return;
      
    // row number of first cell in 
    // current sub-square of size
    // k x k
    for($i = 0; $i < $n - $k + 1; $i++)
    {
          
        // column of first cell in 
        // current sub-square of size
        // k x k
        for($j = 0; $j < $n - $k + 1; $j++)
        {
              
            // Calculate and print sum of
            // current sub-square
            $sum = 0;
            for ($p = $i; $p < $k + $i; $p++)
                for ($q = $j; $q < $k + $j; $q++)
                    $sum += $mat[$p][$q];
            echo $sum , " ";
        }
      
        // Line separator for sub-squares
        // starting with next row
        echo "\n";
    }
}
  
    // Driver Code
    $mat = array(array(1, 1, 1, 1, 1),
                 array(2, 2, 2, 2, 2,),
                  array(3, 3, 3, 3, 3,),
                 array(4, 4, 4, 4, 4,),
                 array(5, 5, 5, 5, 5));
                      
    $k = 3;
    printSumSimple($mat, $k);
  
// This code is contributed by anuj_67.
?>

chevron_right


Output:



  18  18  18
  27  27  27
  36  36  36

Time complexity of above solution is O(k2n2). We can solve this problem in O(n2) time using a Tricky Solution. The idea is to preprocess the given square matrix. In the preprocessing step, calculate sum of all vertical strips of size k x 1 in a temporary square matrix stripSum[][]. Once we have sum of all vertical strips, we can calculate sum of first sub-square in a row as sum of first k strips in that row, and for remaining sub-squares, we can calculate sum in O(1) time by removing the leftmost strip of previous subsquare and adding the rightmost strip of new square.

Following is the implementation of this idea.

C++

filter_none

edit
close

play_arrow

link
brightness_4
code

// An efficient C++ program to find sum of all subsquares of size k x k
#include <iostream>
using namespace std;
  
// Size of given matrix
#define n 5
  
// A O(n^2) function to find sum of all sub-squares of size k x k
// in a given square matrix of size n x n
void printSumTricky(int mat[][n], int k)
{
   // k must be smaller than or equal to n
   if (k > n) return;
  
   // 1: PREPROCESSING
   // To store sums of all strips of size k x 1
   int stripSum[n][n];
  
   // Go column by column
   for (int j=0; j<n; j++)
   {
       // Calculate sum of first k x 1 rectangle in this column
       int sum = 0;
       for (int i=0; i<k; i++)
          sum += mat[i][j];
       stripSum[0][j] = sum;
  
       // Calculate sum of remaining rectangles
       for (int i=1; i<n-k+1; i++)
       {
            sum += (mat[i+k-1][j] - mat[i-1][j]);
            stripSum[i][j] = sum;
       }
   }
  
   // 2: CALCULATE SUM of Sub-Squares using stripSum[][]
   for (int i=0; i<n-k+1; i++)
   {
      // Calculate and print sum of first subsquare in this row
      int sum = 0;
      for (int j = 0; j<k; j++)
           sum += stripSum[i][j];
      cout << sum << "  ";
  
      // Calculate sum of remaining squares in current row by
      // removing the leftmost strip of previous sub-square and
      // adding a new strip
      for (int j=1; j<n-k+1; j++)
      {
         sum += (stripSum[i][j+k-1] - stripSum[i][j-1]);
         cout << sum << "  ";
      }
  
      cout << endl;
   }
}
  
// Driver program to test above function
int main()
{
    int mat[n][n] = {{1, 1, 1, 1, 1},
                     {2, 2, 2, 2, 2},
                     {3, 3, 3, 3, 3},
                     {4, 4, 4, 4, 4},
                     {5, 5, 5, 5, 5},
                    };
    int k = 3;
    printSumTricky(mat, k);
    return 0;
}

chevron_right


Java

filter_none

edit
close

play_arrow

link
brightness_4
code

// An efficient Java program to find
// sum of all subsquares of size k x k
import java.io.*;
  
class GFG {
      
// Size of given matrix
static int n = 5;
  
// A O(n^2) function to find sum of all
// sub-squares of size k x k in a given
// square matrix of size n x n
static void printSumTricky(int mat[][], int k) {
      
    // k must be smaller than or equal to n
    if (k > n)
    return;
  
    // 1: PREPROCESSING
    // To store sums of all strips of size k x 1
    int stripSum[][] = new int[n][n];
  
    // Go column by column
    for (int j = 0; j < n; j++) {
          
    // Calculate sum of first k x 1
    // rectangle in this column
    int sum = 0;
    for (int i = 0; i < k; i++)
        sum += mat[i][j];
    stripSum[0][j] = sum;
  
    // Calculate sum of remaining rectangles
    for (int i = 1; i < n - k + 1; i++) {
        sum += (mat[i + k - 1][j] - mat[i - 1][j]);
        stripSum[i][j] = sum;
    }
    }
  
    // 2: CALCULATE SUM of Sub-Squares 
    // using stripSum[][]
    for (int i = 0; i < n - k + 1; i++) {
          
    // Calculate and print sum of first 
    // subsquare in this row
    int sum = 0;
    for (int j = 0; j < k; j++)
        sum += stripSum[i][j];
    System.out.print(sum + " ");
  
    // Calculate sum of remaining squares 
    // in current row by removing the
    // leftmost strip of previous sub-square 
    // and adding a new strip
    for (int j = 1; j < n - k + 1; j++) {
        sum += (stripSum[i][j + k - 1] - stripSum[i][j - 1]);
        System.out.print(sum + " ");
    }
    System.out.println();
    }
}
  
// Driver program to test above function
public static void main(String[] args)
{
    int mat[][] = {{1, 1, 1, 1, 1},
                   {2, 2, 2, 2, 2},
                   {3, 3, 3, 3, 3},
                   {4, 4, 4, 4, 4}, 
                   {5, 5, 5, 5, 5},
                  };
    int k = 3;
    printSumTricky(mat, k);
}
}
  
// This code is contributed by vt_m.

chevron_right


C#

filter_none

edit
close

play_arrow

link
brightness_4
code

// An efficient C# program to find
// sum of all subsquares of size k x k
using System;
class GFG {
      
    // Size of given matrix
    static int n = 5;
      
    // A O(n^2) function to find sum of all
    // sub-squares of size k x k in a given
    // square matrix of size n x n
    static void printSumTricky(int [,]mat, int k) 
    {
          
        // k must be smaller than or equal to n
        if (k > n)
        return;
      
        // 1: PREPROCESSING
        // To store sums of all strips of
        // size k x 1
        int [,]stripSum = new int[n,n];
      
        // Go column by column
        for (int j = 0; j < n; j++)
        {
              
            // Calculate sum of first k x 1
            // rectangle in this column
            int sum = 0;
            for (int i = 0; i < k; i++)
                sum += mat[i,j];
                  
            stripSum[0,j] = sum;
          
            // Calculate sum of remaining
            // rectangles
            for (int i = 1; i < n - k + 1; i++)
            {
                sum += (mat[i + k - 1,j] 
                               - mat[i - 1,j]);
                stripSum[i,j] = sum;
            }
        }
      
        // 2: CALCULATE SUM of Sub-Squares 
        // using stripSum[][]
        for (int i = 0; i < n - k + 1; i++)
        {
              
            // Calculate and print sum of first 
            // subsquare in this row
            int sum = 0;
            for (int j = 0; j < k; j++)
                sum += stripSum[i,j];
                  
            Console.Write(sum + " ");
          
            // Calculate sum of remaining 
            // squares in current row by 
            // removing the leftmost strip
            // of previous sub-square 
            // and adding a new strip
            for (int j = 1; j < n - k + 1; j++) 
            {
                sum += (stripSum[i,j + k - 1] 
                           - stripSum[i,j - 1]);
                Console.Write(sum + " ");
            }
            Console.WriteLine();
        }
    }
      
    // Driver program to test above function
    public static void Main()
    {
        int [,]mat = {{1, 1, 1, 1, 1},
                    {2, 2, 2, 2, 2},
                    {3, 3, 3, 3, 3},
                    {4, 4, 4, 4, 4}, 
                    {5, 5, 5, 5, 5},
                    };
        int k = 3;
        printSumTricky(mat, k);
    }
}
  
// This code is contributed by nitin mittal.

chevron_right


PHP

filter_none

edit
close

play_arrow

link
brightness_4
code

<?php
// An efficient PHP program to
// find sum of all subsquares
// of size k x k
  
// Size of given matrix
$n = 5;
  
// A O(n^2) function to find 
// sum of all sub-squares of 
// size k x k in a given
// square matrix of size n x n
function printSumTricky($mat, $k)
{
global $n;
  
// k must be smaller 
// than or equal to n
if ($k > $n) return;
  
// 1: PREPROCESSING
// To store sums of all 
// strips of size k x 1
$stripSum = array(array());
  
// Go column by column
for ($j = 0; $j < $n; $j++)
{
    // Calculate sum of first
    // k x 1 rectangle in this column
    $sum = 0;
    for ($i = 0; $i < $k; $i++)
        $sum += $mat[$i][$j];
    $stripSum[0][$j] = $sum;
  
    // Calculate sum of 
    // remaining rectangles
    for ($i = 1; $i < $n - $k + 1; $i++)
    {
            $sum += ($mat[$i + $k - 1][$j] - 
                          $mat[$i - 1][$j]);
            $stripSum[$i][$j] = $sum;
    }
}
  
// 2: CALCULATE SUM of 
// Sub-Squares using stripSum[][]
for ($i = 0; $i < $n - $k + 1; $i++)
{
    // Calculate and print sum of 
    // first subsquare in this row
    $sum = 0;
    for ($j = 0; $j < $k; $j++)
        $sum += $stripSum[$i][$j];
    echo $sum , " ";
  
    // Calculate sum of remaining 
    // squares in current row by
    // removing the leftmost strip 
    // of previous sub-square and
    // adding a new strip
    for ($j = 1; $j < $n - $k + 1; $j++)
    {
        $sum += ($stripSum[$i][$j + $k - 1] - 
                 $stripSum[$i][$j - 1]);
        echo $sum , " ";
    }
  
    echo "\n";
}
}
  
// Driver Code
$mat = array(array(1, 1, 1, 1, 1),
             array(2, 2, 2, 2, 2),
             array(3, 3, 3, 3, 3),
             array(4, 4, 4, 4, 4),
             array(5, 5, 5, 5, 5));
$k = 3;
printSumTricky($mat, $k);
  
// This code is contributed by anuj_67.
?>

chevron_right


Output :

  18  18  18
  27  27  27
  36  36  36

This article is contributed by Rahul Gupta. Please write comments if you find anything incorrect, or you want to share more information about the topic discussed above.



My Personal Notes arrow_drop_up

Improved By : Sam007, nitin mittal, vt_m



Article Tags :
Practice Tags :


2


Please write to us at contribute@geeksforgeeks.org to report any issue with the above content.