Program to Find Out the Number of Squares in a Grid in Python

Suppose we have two values p and q, we have to find the number of unique squares that can be generated from a grid with p rows and q columns in which the points are placed evenly. If the answer is very large return result mod 10^9 + 7. In this problem, a square is a set of 4 points that form the four vertices of a square. The sides of the square must have the same length, and it does not always have the need to be aligned with the axes of the grid.

So, if the input is like p = 4, q = 4, then the output will be 20.

To solve this, we will follow these steps −

  • for i in range minimum of r to c, do,

    • ans := ans +(r - i) *(c - i) * i

    • return ans mod (10^9 + 7)


Let us see the following implementation to get better understanding −

 Live Demo

class Solution:
   def solve(self, r, c):
      ans = 0
      for i in range(min(r, c)):
         ans += (r - i) * (c - i) * i
      return ans % (10 ** 9 + 7)
ob = Solution()


p = 4
q = 4



Updated on: 23-Dec-2020


Kickstart Your Career

Get certified by completing the course

Get Started