Week 1 - Challenge 2 - solution

squares_in_a_grid.py

# 3 x 3 grid, has 1 3*3, 4 2*2, 9 1*1
# the easy way to calculate this is to subtract the number you need from the width
# so 2 x 2 in a 3 x 3 grid would be (3 -(2-1)) * (3 -(2-1)) = 4

gridSize = 9
total = 0
# first we start with the smallest square and go up to the widest square
for x in range(1,gridSize+1):
    if x ==1:
        total = total + (gridSize * gridSize)
    else:
        total = total + (gridSize - (x-1)) * (gridSize - (x-1))
print total