[코드트리] 배열 만들기 > 제곱하여 출력하기

n개의 원소가 주어졌을 때, 주어진 n개의 원소에 각각 제곱을 하여 출력하는 프로그램을 작성해보세요.

 

 

# 변수 선언 및 입력
n = int(input())

# 배열에 주어진 수를 입력받아 저장합니다.
arr = list(map(int, input().split()))

# n개의 정수를 제곱한 배열을 새로 만듭니다.
new_arr = [elem * elem for elem in arr]

# n개의 정수를 출력합니다.
for i in range(n):
	print(new_arr[i], end=" ")
    
# 나의 풀이 ---------------------------------
n = int(input())
arr = list(map(int, input().split()))
new_arr = [i ** 2 for i in arr]

for i in range(len(new_arr)):
    print(new_arr[i], end = " ")