Square_into_Squares_Protect_trees.py

"""
Square into Squares. Protect trees!

http://www.codewars.com/kata/54eb33e5bc1a25440d000891/train/python

"""


def decompose(n):
    goal = 0
    result = [n]
    while result:
        current = result.pop()
        goal += current ** 2
        for i in range(current - 1, 0, -1):
            if goal - (i ** 2) >= 0:
                goal -= i ** 2
                result.append(i)
                if goal == 0:
                    result.sort()
                    return result
    return None