Write a non-recursive function that takes n as parameter and returns the total where total = 1 + 2*2 +3 + 4*4 + (n*n if n even or n if n is odd), i.e. adding odd numbers as well as squares of even numbers to total.

Python please