06. Electron Distribution

numbers_of_electrons = int(input())
list_of = []

for shells in range(1, numbers_of_electrons + 1):
    max_electron_numbers = 2 * shells ** 2
    numbers_of_electrons -= max_electron_numbers
    if numbers_of_electrons >= 0:
        list_of.append(max_electron_numbers)
        if numbers_of_electrons < max_electron_numbers and int(numbers_of_electrons) != 0:
            list_of.append(numbers_of_electrons)
print(list_of)