Необходимо найти значение функции

@droll55 · Регистрация: 12.12.2020

Author24 — интернет-сервис помощи студентам

Доброго всем дня. Прохожу курс по изучению Python. Застрял на подобной задачке:

На вход программе подается натуральное число n, а затем n целых чисел. Напишите программу, которая для каждого введенного числа xx выводит значение функции f(x) = x^2 + 2x + 1 каждое на отдельной строке.

Формат входных данных
На вход программе подаются натуральное число n, а затем n целых чисел, каждое на отдельной строке.

Формат выходных данных
Программа должна вывести сначала введенные числа, затем пустую строку, а затем соответствующие значения функции.

Sample Input:

5
1
2
3
4
5
Sample Output:

1
2
3
4
5

4
9
16
25
36

Мой код:

Python

a = []
b = []
n = int(input())
numbers = []
 
for i in range(1, n + 1):
    print(i)
    
    
print()
 
for i in range(n):
    new_number = ((int(input()) +1) ** 2 )
    numbers.append(new_number)
print (*numbers, sep="\n")
 
# второй вариант решения( тоже не работает)
#for i in range(n):
#    b = [((int(input()) + 1) ** 2)]
#    print(*b)

Код отлично проходит первый тест, но на втором тесте подаются числа 1 и 10. Правильный ответ должен быть: 10 121, когда мой код выдает 1, 121. Подскажите пожалуйста, где у меня ошибка и как решить данную задачу. Спасибо!

@Gdez · 12.12.2020, 16:40

droll55, во втором тесте первым числом должно быть само число -> вывод не "i", a "new_number"

Добавлено через 1 минуту
Даже не "new_number", а то что подается в формуле расчета "new_number"

Добавлено через 2 минуты
Можно так

Python

for i in range(n):
    a.append(int(input()))
    new_number = (a[-1] +1) ** 2 
    numbers.append(new_number)
print(*a, sep='\n')
print()
print (*numbers, sep="\n")

@droll55 · 12.12.2020, 17:21 **[ТС]**

Большое спасибо! Два дня голову ломал, не мог никак решить.

@droll55 2 / 1 / 1 Регистрация: 12.12.2020 Сообщений: 33
	12.12.2020, 17:21 [ТС]	3
	Большое спасибо! Два дня голову ломал, не мог никак решить. 0

Необходимо найти значение функции

Решение